|
51 | 51 | ) %} |
52 | 52 | {% if shard_index == 1 %} |
53 | 53 | {{ 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" |
55 | 55 | // These require a GPU to finish the build (i.e. CUDA needs to be load-able) |
56 | 56 | make_standalone_crt(ci_gpu, 'build') |
57 | 57 | // make_cpp_tests(ci_gpu, 'build') |
|
60 | 60 | sh "rm -rf build" |
61 | 61 | {{ m.download_artifacts(tag='gpu') }} |
62 | 62 | 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" |
64 | 64 | make_standalone_crt(ci_gpu, 'build') |
65 | 65 | make_cpp_tests(ci_gpu, 'build') |
66 | 66 | cpp_unittest(ci_gpu) |
67 | 67 | 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", |
69 | 69 | label: 'Make OpenCL cpp unit tests', |
70 | 70 | ) |
71 | 71 | 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", |
73 | 73 | label: 'Run OpenCL cpp unit tests', |
74 | 74 | ) |
75 | 75 | micro_cpp_unittest(ci_gpu) |
|
79 | 79 | {% endif %} |
80 | 80 | {% if shard_index == 2 or num_shards < 2 %} |
81 | 81 | 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", |
83 | 83 | label: 'Run Java unit tests', |
84 | 84 | ) |
85 | 85 | {% endif %} |
86 | 86 | 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", |
88 | 88 | label: 'Run Python GPU unit tests', |
89 | 89 | ) |
90 | 90 | 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", |
92 | 92 | label: 'Run Python GPU integration tests', |
93 | 93 | ) |
94 | 94 | {% endcall %} |
|
105 | 105 | {{ m.download_artifacts(tag='gpu') }} |
106 | 106 | ci_setup(ci_gpu) |
107 | 107 | 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", |
109 | 109 | label: 'Run TOPI tests', |
110 | 110 | ) |
111 | 111 | {% endcall %} |
|
122 | 122 | {{ m.download_artifacts(tag='gpu') }} |
123 | 123 | ci_setup(ci_gpu) |
124 | 124 | 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", |
126 | 126 | label: 'Run Python frontend tests', |
127 | 127 | ) |
128 | 128 | {% endcall %} |
|
140 | 140 | {{ m.download_artifacts(tag='gpu') }} |
141 | 141 | ci_setup(ci_gpu) |
142 | 142 | 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", |
144 | 144 | label: 'Build docs', |
145 | 145 | ) |
146 | 146 | {{ m.upload_artifacts(tag='docs', filenames=["docs.tgz"]) }} |
|
0 commit comments