def single_ga_tune(self, task_id, json_info):
"""
GA tune for single op
:param task_id: task id for this op to tune
:param json_info: op's info
"""
converted_json = single_to_fusion(json.dumps(json_info),
tune_mode="GA")
graph_id = 0
l1size = 0
kernel_name = json.loads(converted_json)["fusion_op_name"]
dispatch_autotune_task(graph_id, task_id, l1size, converted_json, [],
kernel_name)
def fusion_ga_tune(self, task_id, json_info):
"""
GA tune for fusion op
:param task_id: task id for this op to tune
:param json_info: op's info
"""
if 'fusion_op' not in json_info or not json_info['fusion_op']:
raise ValueError("Json string Errors, key:fusion_op not found.")
kernel_name = json_info["fusion_op"]["fusion_op_name"]
converted_json = fusion_to_fusion(json.dumps(json_info),
tune_mode="GA")
graph_id = 0
l1size = 0
dispatch_autotune_task(graph_id, task_id, l1size, converted_json, [],
kernel_name)