diff --git a/mindspore/_extends/parallel_compile/tbe_compiler/tbe_process.py b/mindspore/_extends/parallel_compile/tbe_compiler/tbe_process.py index 3fe61c4cf9..6ac853cb1c 100644 --- a/mindspore/_extends/parallel_compile/tbe_compiler/tbe_process.py +++ b/mindspore/_extends/parallel_compile/tbe_compiler/tbe_process.py @@ -309,7 +309,7 @@ class TbeProcess: task_future = self.__pool.apply_async(func=run_compiler, args=(op_json,)) self.__running_tasks.append((task_id, task_future)) else: - log.info("start_compile_op: op json:\n {}".format(op_json)) + log.info("start_compile_op: task id: {} op json:\n {}".format(task_id, op_json)) if self.__tuner is None: log.error("Please confirm that the mode isn't NO_TUNE and auto_tune already initialized.") return task_id @@ -376,6 +376,7 @@ class TbeProcess: len(self.__running_tune_tasks), len(self.__all_tune_tasks), len(self.__failed_tune_task))) else: + log.info("get finish tasks:[{}]".format(ret)) for item in ret: task_id = item['task_id'] status_code = item['status_code'] diff --git a/mindspore/_extends/parallel_compile/tbe_compiler/tuner.py b/mindspore/_extends/parallel_compile/tbe_compiler/tuner.py index bfe09915a4..3b886045d8 100644 --- a/mindspore/_extends/parallel_compile/tbe_compiler/tuner.py +++ b/mindspore/_extends/parallel_compile/tbe_compiler/tuner.py @@ -360,7 +360,7 @@ class TbeTuner: converted_json = single_to_fusion(json.dumps(json_info), tune_mode="GA") graph_id = 0 l1size = 0 - kernel_name = json_info["fusion_op_name"] + kernel_name = json.loads(converted_json)["fusion_op_name"] dispatch_autotune_task(graph_id, task_id, l1size, converted_json, [], kernel_name) def get_finish_tasks(self):