Skip to content

Commit c2acc27

Browse files
committed
[CI] Set NVIDIA_DISABLE_REQUIRE=true
1 parent d2c549b commit c2acc27

File tree

2 files changed

+32
-32
lines changed

2 files changed

+32
-32
lines changed

ci/jenkins/generated/gpu_jenkinsfile.groovy

Lines changed: 22 additions & 22 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

ci/jenkins/templates/gpu_jenkinsfile.groovy.j2

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@
5151
) %}
5252
{% if shard_index == 1 %}
5353
{{ m.download_artifacts(tag='gpu2') }}
54-
sh "${docker_run} --no-gpu ${ci_gpu} ./tests/scripts/task_config_build_gpu_other.sh build"
54+
sh "${docker_run} --env NVIDIA_DISABLE_REQUIRE true --no-gpu ${ci_gpu} ./tests/scripts/task_config_build_gpu_other.sh build"
5555
// These require a GPU to finish the build (i.e. CUDA needs to be load-able)
5656
make_standalone_crt(ci_gpu, 'build')
5757
// make_cpp_tests(ci_gpu, 'build')
@@ -60,16 +60,16 @@
6060
sh "rm -rf build"
6161
{{ m.download_artifacts(tag='gpu') }}
6262
ci_setup(ci_gpu)
63-
sh "${docker_run} --no-gpu ${ci_gpu} ./tests/scripts/task_config_build_gpu.sh build"
63+
sh "${docker_run} --env NVIDIA_DISABLE_REQUIRE true --no-gpu ${ci_gpu} ./tests/scripts/task_config_build_gpu.sh build"
6464
make_standalone_crt(ci_gpu, 'build')
6565
make_cpp_tests(ci_gpu, 'build')
6666
cpp_unittest(ci_gpu)
6767
sh (
68-
script: "${docker_run} ${ci_gpu} python3 ./tests/scripts/task_build.py --sccache-bucket tvm-sccache-prod --cmake-target opencl-cpptest --build-dir build",
68+
script: "${docker_run} --env NVIDIA_DISABLE_REQUIRE true ${ci_gpu} python3 ./tests/scripts/task_build.py --sccache-bucket tvm-sccache-prod --cmake-target opencl-cpptest --build-dir build",
6969
label: 'Make OpenCL cpp unit tests',
7070
)
7171
sh (
72-
script: "${docker_run} ${ci_gpu} ./tests/scripts/task_opencl_cpp_unittest.sh",
72+
script: "${docker_run} --env NVIDIA_DISABLE_REQUIRE true ${ci_gpu} ./tests/scripts/task_opencl_cpp_unittest.sh",
7373
label: 'Run OpenCL cpp unit tests',
7474
)
7575
micro_cpp_unittest(ci_gpu)
@@ -79,16 +79,16 @@
7979
{% endif %}
8080
{% if shard_index == 2 or num_shards < 2 %}
8181
sh (
82-
script: "${docker_run} ${ci_gpu} ./tests/scripts/task_java_unittest.sh",
82+
script: "${docker_run} --env NVIDIA_DISABLE_REQUIRE true ${ci_gpu} ./tests/scripts/task_java_unittest.sh",
8383
label: 'Run Java unit tests',
8484
)
8585
{% endif %}
8686
sh (
87-
script: "${docker_run} ${ci_gpu} ./tests/scripts/task_python_unittest_gpuonly.sh",
87+
script: "${docker_run} --env NVIDIA_DISABLE_REQUIRE true ${ci_gpu} ./tests/scripts/task_python_unittest_gpuonly.sh",
8888
label: 'Run Python GPU unit tests',
8989
)
9090
sh (
91-
script: "${docker_run} ${ci_gpu} ./tests/scripts/task_python_integration_gpuonly.sh",
91+
script: "${docker_run} --env NVIDIA_DISABLE_REQUIRE true ${ci_gpu} ./tests/scripts/task_python_integration_gpuonly.sh",
9292
label: 'Run Python GPU integration tests',
9393
)
9494
{% endcall %}
@@ -105,7 +105,7 @@
105105
{{ m.download_artifacts(tag='gpu') }}
106106
ci_setup(ci_gpu)
107107
sh (
108-
script: "${docker_run} ${ci_gpu} ./tests/scripts/task_python_topi.sh",
108+
script: "${docker_run} --env NVIDIA_DISABLE_REQUIRE true ${ci_gpu} ./tests/scripts/task_python_topi.sh",
109109
label: 'Run TOPI tests',
110110
)
111111
{% endcall %}
@@ -122,7 +122,7 @@
122122
{{ m.download_artifacts(tag='gpu') }}
123123
ci_setup(ci_gpu)
124124
sh (
125-
script: "${docker_run} ${ci_gpu} ./tests/scripts/task_python_frontend.sh",
125+
script: "${docker_run} --env NVIDIA_DISABLE_REQUIRE true ${ci_gpu} ./tests/scripts/task_python_frontend.sh",
126126
label: 'Run Python frontend tests',
127127
)
128128
{% endcall %}
@@ -140,7 +140,7 @@
140140
{{ m.download_artifacts(tag='gpu') }}
141141
ci_setup(ci_gpu)
142142
sh (
143-
script: "${docker_run} ${ci_gpu} ./tests/scripts/task_python_docs.sh",
143+
script: "${docker_run} --env NVIDIA_DISABLE_REQUIRE ${ci_gpu} ./tests/scripts/task_python_docs.sh",
144144
label: 'Build docs',
145145
)
146146
{{ m.upload_artifacts(tag='docs', filenames=["docs.tgz"]) }}

0 commit comments

Comments
 (0)