def __init__(self):
self.id_ = oneflow_api.NewSessionId()
self.job_name2function_desc_ = {}
self.status_ = SessionStatus.OPEN
self.cond_var_ = threading.Condition()
self.running_job_cnt_ = 0
self.inter_user_job_info_ = None
self.uuid2watch_handler_ = {}
self.config_proto_ = None
self.resource_ = None
self.is_mirrored_strategy_enabled_stack_ = []
self.job_name2var_name2var_blob_ = {}
self.job_name2module_name2module_ = {}
self.existed_module_names_ = set()
self.var_name2var_blob_ = {}
# parallel desc symbol id in op attribute does not always correct
# for lazy ops as parallel conf may be updated in some passes
# (like optimizer_placement_optimization_pass)
self.interface_op_name2op_attr_ = {}
self.interface_op_name2job_name_ = {}
self.lazy_interface_op_name2parallel_conf_ = {}
self.op_name2lazy_blob_cache_ = {}
self.job_name2name_scope_stack_ = {}
self.eager_global_function_desc_stack_ = []
self.function_flag_name2default_val_ = {}
self._UpdateFunctionFlagName2DefaultVal()
self.scope_attr_name2default_val_ = {}
self._UpdateScopeAttrName2DefaultVal()
self.instruction_list_ = instr_cfg.InstructionListProto()
self.eager_symbol_list_ = eager_symbol_util.EagerSymbolList()
self.backward_blob_register_ = blob_register_util.BlobRegister()
self.snapshot_mgr_ = SnapshotManager()
self.eager_config_proto_ctx_ = None
def __init__(self):
self.id_ = oneflow_api.NewSessionId()
self.job_name2function_desc_ = {}
self.status_ = SessionStatus.OPEN
self.cond_var_ = threading.Condition()
self.running_job_cnt_ = 0
self.inter_user_job_info_ = None
self.uuid2watch_handler_ = {}
self.config_proto_ = None
self.resource_ = None
self.is_mirrored_strategy_enabled_stack_ = []
self.function_flag_name2default_val_ = {}
self.job_name2var_name2var_blob_ = {}
self.job_name2module_name2module_ = {}
self.existed_module_names_ = set()
self.var_name2var_blob_ = {}
self.interface_op_name2op_attr_ = {}
self.interface_op_name2job_name_ = {}
self.job_name2name_scope_stack_ = {}
self.eager_global_function_desc_stack_ = []
self._UpdateFunctionFlagName2DefaultVal()
self.instruction_list_ = instr_util.InstructionListProto()
self.eager_symbol_list_ = eager_symbol_util.EagerSymbolList()
self.backward_blob_register_ = blob_register_util.BlobRegister()
self.snapshot_mgr_ = SnapshotManager()
self.eager_config_proto_ctx_ = None
def clear_default_session():
session_ctx.TryCloseDefaultSession()
session_ctx.OpenDefaultSession(Session(oneflow_api.NewSessionId()))
func = enable_if.unique([sync_default_session])
return func()
@enable_if.condition(hob.in_normal_mode)
def sync_default_session() -> None:
session_ctx.GetDefaultSession().Sync()
def _TryCompleteConfigProto(config_proto):
if config_proto.resource.machine_num == 0:
config_proto.resource.machine_num = len(
env_util.default_env_proto.machine)
def _GetDefaultConfigProto():
config_proto = job_set_util.ConfigProto()
config_proto.resource.machine_num = 0
if oneflow_api.flags.with_cuda():
config_proto.resource.gpu_device_num = 1
else:
config_proto.resource.cpu_device_num = 1
config_proto.resource.gpu_device_num = 0
config_proto.io_conf.data_fs_conf.localfs_conf.SetInParent()
config_proto.io_conf.snapshot_fs_conf.localfs_conf.SetInParent()
config_proto.session_id = session_ctx.GetDefaultSession().id
return config_proto
session_ctx.OpenDefaultSession(Session(oneflow_api.NewSessionId()))
