| /third_party/mindspore/mindspore/_extends/parallel_compile/tbe_compiler/ |
| D | tuner.py | 46 def __init__(self, offline_tune, tune_mode): argument
54 self._creating_custom_path(tune_mode)
70 tune_mode = self.get_tune_mode(json_info)
71 ret = self.parallel_compilation_init(soc_info, tune_mode, process_num)
93 tune_mode = json_info["SocInfo"]["autoTilingMode"]
95 tune_mode = "RL"
96 return tune_mode
107 def __creating_default_custom_path(self, tune_mode, base_custom_path): argument
113 if "RL" in tune_mode:
115 if "GA" in tune_mode:
[all …]
|
| D | tbe_process.py | 146 self.tune_mode = None
183 def init_auto_tune_env(self, tune_mode): argument
189 self.tune_mode = tune_mode
193 if self.tune_mode == NO_TUNE and not self.offline_tune:
221 self.__tuner = TbeTuner(self.offline_tune, self.tune_mode)
276 tune_mode = json_info["SocInfo"]["autoTilingMode"]
283 tune_mode = NO_TUNE
284 if GA_TUNE in tune_mode:
288 if RL_TUNE in tune_mode:
336 tune_mode = self.select_tune_mode(json_info)
[all …]
|
| D | re_construct_json.py | 125 def single_to_fusion(json_file, tune_mode): argument
135 soc_info["autoTilingMode"] = tune_mode
159 def fusion_to_fusion(json_str, tune_mode): argument
169 json_info["SocInfo"]["autoTilingMode"] = tune_mode
|
| D | compiler.py | 56 def build_op(build_type, json_str, tune_mode=None): argument
132 if tune_mode is not None:
155 if tune_mode is not None:
|
| D | tbe_job_manager.py | 215 tune_mode = self._select_tune_mode(job)
216 if tune_mode == TuneMode.NO_TUNE:
220 if tune_mode == TuneMode.RL_TUNE:
234 if tune_mode == TuneMode.RL_TUNE:
240 … "Tune Job failed, tune type {}, job json string:{}".format(tune_mode, job.json_string))
243 if tune_mode == TuneMode.RL_TUNE and len(compute_op_list) == 1:
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_fusion.cc | 63 auto tune_mode = context_ptr->get_param<std::string>(MS_CTX_TUNE_MODE); in KernelFusion() local
96 …if (kernel_pack != nullptr && ((!offline_tune.empty() && offline_tune != "true") || tune_mode == "… in KernelFusion()
112 soc_info_json["autoTilingMode"] = tune_mode; in KernelFusion()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_parallel_build.cc | 44 auto tune_mode = context_ptr->get_param<std::string>(MS_CTX_TUNE_MODE); in TbeOpParallelBuild() local
52 << tune_mode; in TbeOpParallelBuild()
76 ((!offline_tune.empty() && offline_tune != "true") || tune_mode == "NO_TUNE")) { in TbeOpParallelBuild()
300 auto tune_mode = kernel_json["SocInfo"]["autoTilingMode"]; in StartCompileOp() local
301 return AscendKernelBuildClient::Instance().TbeStart(kernel_json.dump(), tune_mode); in StartCompileOp()
|
| D | tbe_kernel_build.cc | 230 auto tune_mode = context_ptr->get_param<std::string>(MS_CTX_TUNE_MODE); in GenTbeSingleKernelJson() local
233 soc_info_json[kAutoTilingMode] = tune_mode; in GenTbeSingleKernelJson()
|
| /third_party/gstreamer/gstplugins_bad/sys/msdk/ |
| D | gstmsdkh265enc.c | 534 encoder->param.mfx.LowPower = h265enc->tune_mode; in gst_msdkh265enc_configure()
677 thiz->tune_mode = in gst_msdkh265enc_set_property()
696 thiz->tune_mode = g_value_get_enum (value); in gst_msdkh265enc_set_property()
765 g_value_set_enum (value, thiz->tune_mode); in gst_msdkh265enc_get_property()
844 h265enc->tune_mode == MFX_CODINGOPTION_OFF) in gst_msdkh265enc_need_conversion()
975 thiz->tune_mode = PROP_TUNE_MODE_DEFAULT; in gst_msdkh265enc_init()
|
| D | gstmsdkh264enc.c | 385 encoder->param.mfx.LowPower = thiz->tune_mode; in gst_msdkh264enc_configure()
578 thiz->tune_mode = in gst_msdkh264enc_set_property()
599 thiz->tune_mode = g_value_get_enum (value); in gst_msdkh264enc_set_property()
660 g_value_set_enum (value, thiz->tune_mode); in gst_msdkh264enc_get_property()
825 thiz->tune_mode = PROP_TUNE_MODE_DEFAULT; in gst_msdkh264enc_init()
|
| D | gstmsdkh265enc.h | 63 gint tune_mode; member
|
| D | gstmsdkh264enc.h | 72 gint tune_mode; member
|
| /third_party/mindspore/mindspore/ |
| D | context.py | 210 def set_auto_tune_mode(self, tune_mode): argument
212 if tune_mode in candidate:
213 self.set_param(ms_ctx_param.tune_mode, tune_mode)
|
| /third_party/mindspore/mindspore/_extends/remote/ |
| D | kernel_build_server_ascend.py | 77 tune_mode = self.get_message()
78 ans = self.tbe_builder.init_auto_tune_env(tune_mode)
|