leandron commented on a change in pull request #7304:
URL: https://github.com/apache/tvm/pull/7304#discussion_r572101488
##########
File path: python/tvm/driver/tvmc/compiler.py
##########
@@ -191,22 +198,21 @@ def compile_model(
if use_autoscheduler:
with auto_scheduler.ApplyHistoryBest(tuning_records):
- with tvm.transform.PassContext(
- opt_level=3, config={"relay.backend.use_auto_scheduler":
True}
- ):
+ config["relay.backend.use_auto_scheduler"] = True
+ with tvm.transform.PassContext(opt_level=3, config=config):
Review comment:
When arguments are passed to the BYOC target e.g. `ethos-n77
-option=value`, they populate this `config` dict, to be passed to the
PassContext. In that way, we should expect `config` to exist at this point, and
want to extend it with `relay.backend.use_auto_scheduler` in this case.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
