diff --git a/python/tvm/contrib/hexagon/session.py b/python/tvm/contrib/hexagon/session.py index da35477e1c..2b3bc2c764 100644 --- a/python/tvm/contrib/hexagon/session.py +++ b/python/tvm/contrib/hexagon/session.py @@ -382,8 +382,8 @@ def _relax_vm_executable_executor( hexagon_arch="v68", ) - self.upload(path_exec, "exec.so") - return self._rpc.get_function("tvm.hexagon.load_module")("exec.so") + path = self.upload(path_exec, "exec.so") + return self._rpc.get_function("tvm.hexagon.load_module")(str(path)) def _aot_executor_from_factory( self, diff --git a/tests/scripts/task_python_hexagon.sh b/tests/scripts/task_python_hexagon.sh index 074c1d5de9..abbc287a98 100755 --- a/tests/scripts/task_python_hexagon.sh +++ b/tests/scripts/task_python_hexagon.sh @@ -50,7 +50,14 @@ fi export ANDROID_SERIAL_NUMBER=${device_serial} # Only test integration with Relax -run_pytest ctypes python-contrib-hexagon tests/python/contrib/test_hexagon/test_relax_integration.py +# TODO(prakalp): Run the same tests on simulator and device once the bug with device is fixed. +if [[ "${device_serial}" == "simulator" ]]; + then + run_pytest ctypes python-contrib-hexagon tests/python/contrib/test_hexagon/test_relax_integration.py + else + run_pytest ctypes python-contrib-hexagon tests/python/contrib/test_hexagon/test_relax_integration.py::test_conv2d +fi + if [[ "${device_serial}" == "simulator" ]]; then kill ${TRACKER_PID}