flame.oneflow.helpers.rank0#

Module Contents#

Classes#

FakeObject

Functions#

do_nothing(*args, **kwargs)

rank0(func)

run func only on rank 0

Attributes#

T

class flame.oneflow.helpers.rank0.FakeObject#
__getattr__(_name)#
Parameters:

_name (str) –

Return type:

Callable

__bool__()#
Return type:

bool

flame.oneflow.helpers.rank0.do_nothing(*args, **kwargs)#
Return type:

FakeObject

flame.oneflow.helpers.rank0.T#
flame.oneflow.helpers.rank0.rank0(func)#

run func only on rank 0

You can use it as a decorator

@rank0
def my_print(*args, **kwargs):
    print(*args, **kwargs)
Parameters:

func (T) – function or lambda

Return type:

T