From 4589ea9d4bac09ce320a101be9e19fda22e3ba09 Mon Sep 17 00:00:00 2001 From: Peter Salas Date: Thu, 9 Sep 2021 15:17:16 -0700 Subject: [PATCH] [Autoscheduler] Reduce task weight coercion overhead --- python/tvm/auto_scheduler/relay_integration.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/auto_scheduler/relay_integration.py b/python/tvm/auto_scheduler/relay_integration.py index 8b68f4e9002a..1977d3de5506 100644 --- 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: