Home
last modified time | relevance | path

Searched refs:auto_tiling_mode (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/_extends/parallel_compile/tbe_compiler/
Dtbe_adapter.py113 def __creating_default_custom_path(auto_tiling_mode, base_custom_path): argument
119 if "RL" in auto_tiling_mode:
121 if "GA" in auto_tiling_mode:
136 auto_tiling_mode = job.content["SocInfo"]["autoTilingMode"]
137 if "NO_TUNE" in auto_tiling_mode:
156 return __creating_default_custom_path(auto_tiling_mode, base_custom_path)
169 auto_tiling_mode = initialize.content["SocInfo"]["autoTilingMode"]
176 ret = init_multi_process_env(embedding, soc_info, auto_tiling_mode, "0",
182 if "RL" in auto_tiling_mode or offline_tune:
193 if "GA" in auto_tiling_mode:
[all …]
Dtbe_job_manager.py50 self.auto_tiling_mode = ""
79 self.auto_tiling_mode = ""
89 tbe_finalize(self.auto_tiling_mode, self.offline_tune)
140 if "GA" in self.auto_tiling_mode:
149 res = tbe_finalize(self.auto_tiling_mode, self.offline_tune)
308 self.auto_tiling_mode = initialize_job.content["SocInfo"]["autoTilingMode"]
341 auto_tiling_mode = job.content["SocInfo"]["autoTilingMode"]
349 if TuneMode.GA_TUNE.value in auto_tiling_mode:
353 if TuneMode.RL_TUNE.value in auto_tiling_mode:
Dcompiler.py146 auto_tiling_mode=None,
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dascend_kernel_compile.cc756 auto auto_tiling_mode = (init_json[kJobContent][kSocInfo]["autoTilingMode"]).get<std::string>(); in TbeInitialize() local
758 is_tune_flag_ = offline_tune || (auto_tiling_mode != "NO_TUNE"); in TbeInitialize()