diff --git a/tests/scripts/task_python_topi.sh b/tests/scripts/task_python_topi.sh index 9a5991e6a766..06046ad76161 100755 --- a/tests/scripts/task_python_topi.sh +++ b/tests/scripts/task_python_topi.sh @@ -21,6 +21,8 @@ set -u source tests/scripts/setup-pytest-env.sh +export TVM_TEST_TARGETS="llvm; llvm -device=arm_cpu; cuda; cuda -model=unknown -libs=cudnn" + # to avoid CI thread throttling. export TVM_BIND_THREADS=0 export OMP_NUM_THREADS=1