Mousius commented on a change in pull request #9282:
URL: https://github.com/apache/tvm/pull/9282#discussion_r730844910
##########
File path: python/tvm/relay/backend/te_compiler.py
##########
@@ -380,88 +375,32 @@ def jit(self, source_func, target=None):
jited_func: tvm.runtime.PackedFunc
The result of jited function.
"""
+ print("caling jit \n")
Review comment:
Missed a print.
##########
File path: tests/python/relay/aot/aot_test_utils.py
##########
@@ -721,7 +723,7 @@ def compile_and_run(
def generate_ref_data(mod, input_data, params=None, target="llvm"):
"""Generate reference data through executing the relay module"""
- compile_engine.get().clear()
+ te_compiler.get().clear()
Review comment:
Do we even need this anymore for AOT?
##########
File path: tests/python/contrib/test_vitis_ai/infrastructure.py
##########
@@ -18,24 +18,23 @@
"""Expose Vitis-AI test functions to the Python frontend"""
+from tvm.contrib import utils
+from tvm.contrib import graph_executor
+from tvm.contrib.target import vitis_ai
+from tvm.relay.build_module import bind_params_by_name
+from tvm.relay.op.contrib.vitis_ai import partition_for_vitis_ai
+from tvm.relay import transform
+from tvm import runtime
+from tvm import relay
+import tvm
+import pyxir.contrib.target.DPUCZDX8G
+import pyxir.contrib.target.DPUCADX8G
import sys
Review comment:
Black shouldn't have done this as it mutates the AST? Potentially
something that imports isort?
--
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.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
