[Autoscheduler] Reduce task weight coercion overhead (#8995)

petersalas · Peter Salas · web-flow · commit 92903b45659f · 2021-09-14T10:44:59.000+09:00
Co-authored-by: Peter Salas &lt;psalas@octoml.ai&gt;
diff --git a/python/tvm/auto_scheduler/relay_integration.py b/python/tvm/auto_scheduler/relay_integration.py
@@ -171,7 +171,7 @@ def extract_tasks(
                 desc=",".join(func_names),
             )
         )
-        weights.append(weight)
+        weights.append(int(weight))
 
     if dump_workload_to_dag_log is not None:
         with open(dump_workload_to_dag_log, "w") as f:

Original file line number	Diff line number	Diff line change
`@@ -171,7 +171,7 @@ def extract_tasks(`
`171`	`171`	`desc=",".join(func_names),`
`172`	`172`	`)`
`173`	`173`	`)`
`174`		`- weights.append(weight)`
	`174`	`+ weights.append(int(weight))`
`175`	`175`
`176`	`176`	`if dump_workload_to_dag_log is not None:`
`177`	`177`	`with open(dump_workload_to_dag_log, "w") as f:`