!12978 fix ga single op tune

From: @laiyongqiang
Reviewed-by: @kisnwang,@zhoufeng54
Signed-off-by: @zhoufeng54
pull/12978/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit c1422cefec

@ -309,7 +309,7 @@ class TbeProcess:
task_future = self.__pool.apply_async(func=run_compiler, args=(op_json,)) task_future = self.__pool.apply_async(func=run_compiler, args=(op_json,))
self.__running_tasks.append((task_id, task_future)) self.__running_tasks.append((task_id, task_future))
else: 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: if self.__tuner is None:
log.error("Please confirm that the mode isn't NO_TUNE and auto_tune already initialized.") log.error("Please confirm that the mode isn't NO_TUNE and auto_tune already initialized.")
return task_id return task_id
@ -376,6 +376,7 @@ class TbeProcess:
len(self.__running_tune_tasks), len(self.__running_tune_tasks),
len(self.__all_tune_tasks), len(self.__failed_tune_task))) len(self.__all_tune_tasks), len(self.__failed_tune_task)))
else: else:
log.info("get finish tasks:[{}]".format(ret))
for item in ret: for item in ret:
task_id = item['task_id'] task_id = item['task_id']
status_code = item['status_code'] status_code = item['status_code']

@ -360,7 +360,7 @@ class TbeTuner:
converted_json = single_to_fusion(json.dumps(json_info), tune_mode="GA") converted_json = single_to_fusion(json.dumps(json_info), tune_mode="GA")
graph_id = 0 graph_id = 0
l1size = 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) dispatch_autotune_task(graph_id, task_id, l1size, converted_json, [], kernel_name)
def get_finish_tasks(self): def get_finish_tasks(self):

Loading…
Cancel
Save