Searched refs:soc_info (Results 1 – 5 of 5) sorted by relevance
/third_party/mindspore/mindspore/_extends/parallel_compile/tbe_compiler/ |
D | tuner.py | 66 soc_info = self.get_soc_info(json_info) 67 cur_cce_product_params = te_set_version(*soc_info) 71 ret = self.parallel_compilation_init(soc_info, tune_mode, process_num) 163 soc_info = [] 164 soc_info.append(json_info["SocInfo"]["socVersion"]) 165 soc_info.append(json_info["SocInfo"]["coreType"]) 166 soc_info.append(json_info["SocInfo"]["coreNum"]) 167 soc_info.append(json_info["SocInfo"]["l1Fusion"]) 168 soc_info.append(json_info["SocInfo"]["l2Mode"]) 169 soc_info.append(json_info["SocInfo"]["l2Fusion"]) [all …]
|
D | tbe_helper.py | 65 soc_info = list() 66 soc_info.append(initialize_job_info["SocInfo"]["socVersion"]) 67 soc_info.append(initialize_job_info["SocInfo"]["coreType"]) 68 soc_info.append(initialize_job_info["SocInfo"]["coreNum"]) 69 soc_info.append(initialize_job_info["SocInfo"]["l1Fusion"]) 70 soc_info.append(initialize_job_info["SocInfo"]["l2Mode"]) 71 soc_info.append(initialize_job_info["SocInfo"]["l2Fusion"]) 72 soc_info.append(soc_param) 74 return soc_info
|
D | re_construct_json.py | 134 soc_info = ori_file["SocInfo"] 135 soc_info["autoTilingMode"] = tune_mode 141 "SocInfo": soc_info,
|
D | tbe_adapter.py | 168 soc_info = get_soc_info(initialize.content) 176 ret = init_multi_process_env(embedding, soc_info, auto_tiling_mode, "0", 188 ret = rl_tune_init(soc_info, res_queue, live_checker, termin_event, log_level, pid_ts) 206 soc_info = get_soc_info(job.content) 207 res = te_set_version(*soc_info)
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
D | ascend_kernel_compile.cc | 752 nlohmann::json soc_info = TbeUtils::GenSocInfo(); in TbeInitialize() local 753 JsonAssemble(kInitialize, soc_info, &init_json); in TbeInitialize()
|