Support ahb from tune_relay.

apache · Jun 29, 2022 · fb2e184 · fb2e184
1 parent 2e7c6b7
commit fb2e184
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/tvm/meta_schedule/tune.py b/python/tvm/meta_schedule/tune.py
@@ -347,8 +347,9 @@ def tune_extracted_tasks(
         cost_model=cost_model,
         measure_callbacks=measure_callbacks,
     )
-    task_scheduler.tune()
-    cost_model.save(osp.join(work_dir, "cost_model.xgb"))
+    if config.max_trials_global > 0:
+        task_scheduler.tune()
+        cost_model.save(osp.join(work_dir, "cost_model.xgb"))
     return database