tqchen commented on a change in pull request #6903:
URL: https://github.com/apache/incubator-tvm/pull/6903#discussion_r524453803
##########
File path: python/tvm/auto_scheduler/relay_integration.py
##########
@@ -34,18 +34,26 @@
def call_all_topi_funcs(mod, params, target):
- """Call all TOPI compute + schedule to extract tasks in a relay program"""
+ """Call all TOPI compute to extract auto_scheduler tasks in a Relay
program"""
# pylint: disable=import-outside-toplevel
from tvm import relay
from tvm.relay.backend import graph_runtime_codegen
- with transform.PassContext(opt_level=3):
+ # Turn off AutoTVM config not found warnings
+ old_autotvm_silent = autotvm.GLOBAL_SCOPE.silent
+ autotvm.GLOBAL_SCOPE.silent = True
+
+ with transform.PassContext(opt_level=3,
config={"relay.backend.use_auto_scheduler": True}):
Review comment:
I see, I think we can go with the current proposal if that is the concern
----------------------------------------------------------------
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]
