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)