예제 #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
파일: env_util.py 프로젝트: lsf1001/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)
    if oneflow_api.CurrentMachineId() == 0:
        scope_util.InitScopeStack()
    else:
        exit(0)
    return True