[Unity] Consider target context for Relay to Relax conversion (#14269)

Some of Relay passes are target specific (for example, AlterOpLayout). This commit adds target information for Relay -> Relax conversion.
apache · Apr 1, 2023 · 96cd5b5 · 96cd5b5
1 parent 08e2a69
commit 96cd5b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relax/testing/relay_translator.py b/python/tvm/relax/testing/relay_translator.py
@@ -236,7 +236,7 @@ def visit_func(node):
 
     # Since optimization passes and OpStrategy are highly context-dependent,
     # we match the exact same context with `extract_task_from_relay()` env
-    with _autotvm_silencer(), tvm.transform.PassContext(
+    with target, _autotvm_silencer(), tvm.transform.PassContext(
         opt_level=opt_level,
         config=pass_config,
         disabled_pass=disabled_pass,