tqchen commented on a change in pull request #6331:
URL: https://github.com/apache/incubator-tvm/pull/6331#discussion_r476625703
##########
File path: tests/scripts/setup-pytest-env.sh
##########
@@ -26,5 +26,20 @@ else
fi
set -u
+export TVM_TEST_DEVICES=""
+while test $# -gt 0
+do
+ case "$1" in
+ cpu) export TVM_TEST_DEVICES="llvm;llvm
-device=arm_cpu;$TVM_TEST_DEVICES"
+ ;;
Review comment:
The easiest way is to just use another script e.g.
task_integration_gpuonly.sh where we setup the env variable there before source
the `task_integration.sh`
----------------------------------------------------------------
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]
