Skip to content

Commit

Permalink
[UnitTest] Removed vulkan from CI run of task_python_topi.sh (apache#…
Browse files Browse the repository at this point in the history
…9219)

Vulkan unit tests were enabled with
apache#9093, but were only intended to run
tests/python/unittest/test_target_codegen_vulkan.py.  Since
task_python_topi.sh did not explicitly specify `TVM_TEST_TARGETS`, it
defaulted to `tvm.testing.utils.DEFAULT_TEST_TARGETS`, which includes
the vulkan runtime.

This commit adds an explicit specification of `TVM_TEST_TARGETS` for
the topi unit tests, matching the targets enabled in the CI GPU build,
but excluding vulkan.
  • Loading branch information
Lunderberg authored and ylc committed Jan 13, 2022
1 parent d9348ee commit 3133d2c
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions tests/scripts/task_python_topi.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 3133d2c

Please sign in to comment.