示例#1
0
文件： env_util.py 项目： zyg11/oneflow 
def env_init():
    global default_env_proto
    assert len(default_env_proto.machine) > 0
    CompleteEnvProto(default_env_proto)
    c_api_util.InitEnv(default_env_proto)
    scope_util.InitScopeStack()
    return True
示例#2
0
def env_init():
    global default_env_proto
    assert len(default_env_proto.machine) > 0
    CompleteEnvProto(default_env_proto)
    c_api_util.InitEnv(default_env_proto)
    if oneflow_api.CurrentMachineId() == 0:
        scope_util.InitScopeStack()
    else:
        exit(0)
    return True