tristan-arm commented on a change in pull request #6982:
URL: https://github.com/apache/tvm/pull/6982#discussion_r574625796
##########
File path: tests/scripts/task_python_integration.sh
##########
@@ -55,6 +55,9 @@ TVM_FFI=ctypes python3 -m pytest apps/dso_plugin_module
# TVM_FFI=ctypes sh prepare_and_test_tfop_module.sh
TVM_FFI=ctypes python3 -m pytest tests/python/integration
+if python -c "import tvm; from tvm.relay.op.contrib.ethosn import
ethosn_available; print(ethosn_available().name)" -eq "SW_ONLY"; then
+ ETHOSN_VARIANT_CONFIG=ETHOSN78_1TOPS_4PLE_448KSRAM TVM_FFI=ctypes python3 -m
pytest tests/python/contrib/test_ethosn
Review comment:
Not needed for the test suite you identified as that one's not part of
the docker image and already targets the ethosn tests specifically - It
defaults to Ethos-N77 but setting ETHOSN_VARIANT_CONFIG appropriately switches
that to Ethos-N78
----------------------------------------------------------------
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]
