Skip to content

Commit

Permalink
[CI] Remove llvm -device=arm_cpu and cuda -libs=cudnn from the de…
Browse files Browse the repository at this point in the history
…fault test target list (#10500)

After recent improvement in GPU frontend tests, I found that `topi: GPU` has become a bottleneck. From the log https://ci.tlcpack.ai/blue/organizations/jenkins/tvm/detail/PR-10391/14/pipeline/319, it is clear that topi tests are running on target `llvm -device=arm_cpu` and `cuda -libs=cudnn`, which I claim is completely redundant since we already run on `llvm` and `cuda` targets. 

In #9905, I've already removed them from `DEFAULT_TEST_TARGETS`, but `topi: GPU` uses its own list of targets which still includes `llvm -device=arm_cpu` and `cuda -libs=cudnn`. I propose to remove them from topi test targets, which hopefully will cut topi GPU tests time by half.
  • Loading branch information
masahi authored Mar 6, 2022
1 parent 085d36c commit 96069c1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/scripts/task_python_topi.sh
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ set -u

source tests/scripts/setup-pytest-env.sh

export TVM_TEST_TARGETS="llvm; llvm -device=arm_cpu; cuda; cuda -model=unknown -libs=cudnn"
export TVM_TEST_TARGETS="llvm; cuda"

# to avoid CI thread throttling.
export TVM_BIND_THREADS=0
Expand Down

0 comments on commit 96069c1

Please sign in to comment.