Browse Source

!12978 fix ga single op tune

From: @laiyongqiang
Reviewed-by: @kisnwang,@zhoufeng54
Signed-off-by: @zhoufeng54
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
c1422cefec
2 changed files with 3 additions and 2 deletions
  1. +2
    -1
      mindspore/_extends/parallel_compile/tbe_compiler/tbe_process.py
  2. +1
    -1
      mindspore/_extends/parallel_compile/tbe_compiler/tuner.py

+ 2
- 1
mindspore/_extends/parallel_compile/tbe_compiler/tbe_process.py View File

@@ -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']


+ 1
- 1
mindspore/_extends/parallel_compile/tbe_compiler/tuner.py View File

@@ -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):


Loading…
Cancel
Save