Skip to content

Commit 0e8ae1e

Browse files
committed
Differentiate pre-build script during build and test
1 parent 1f44655 commit 0e8ae1e

File tree

3 files changed

+5
-2
lines changed

3 files changed

+5
-2
lines changed

.github/workflows/build-test-windows.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,7 @@ jobs:
172172
ref: ""
173173
test-infra-repository: gs-olive/test-infra
174174
test-infra-ref: windows_builds
175+
pre-script: packaging/pre_build_script_driver_upgrade.sh
175176
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
176177
script: |
177178
export USE_HOST_DEPS=1
Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
curl -so cuda_12.1.0_windows_network.exe https://developer.download.nvidia.com/compute/cuda/12.1.0/network_installers/cuda_12.1.0_windows_network.exe
2+
./cuda_12.1.0_windows_network.exe -s nvcc_12.1 Display.Driver -n

packaging/pre_build_script_windows.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,6 @@ cat ./toolchains/ci_workspaces/WORKSPACE.win.release.tmpl | envsubst > WORKSPACE
1313

1414
# Adapted from:
1515
# https://github.com/HolyWu/TensorRT/commit/41237cedd56809062ad8e29ed0a18fabd54dd3e1#diff-97ebcd21a0f2f4e262f4e6c703e13e8cc6b233f9d9861eb2d9d06994f555f144
16-
curl -so cuda_12.1.0_windows_network.exe https://developer.download.nvidia.com/compute/cuda/12.1.0/network_installers/cuda_12.1.0_windows_network.exe
17-
./cuda_12.1.0_windows_network.exe -s nvcc_12.1 Display.Driver -n
16+
# curl -so cuda_12.1.0_windows_network.exe https://developer.download.nvidia.com/compute/cuda/12.1.0/network_installers/cuda_12.1.0_windows_network.exe
17+
# ./cuda_12.1.0_windows_network.exe -s nvcc_12.1 Display.Driver -n
1818
# ./cuda_12.1.0_windows_network.exe -s cuda_profiler_api_12.1 cudart_12.1 cuobjdump_12.1 cupti_12.1 cuxxfilt_12.1 nvcc_12.1 nvdisasm_12.1 nvjitlink_12.1 nvml_dev_12.1 nvprof_12.1 nvprune_12.1 nvrtc_12.1 nvrtc_dev_12.1 nvtx_12.1 nvvm_samples_12.1 opencl_12.1 visual_profiler_12.1 sanitizer_12.1 thrust_12.1 cublas_12.1 cublas_dev_12.1 cufft_12.1 cufft_dev_12.1 curand_12.1 curand_dev_12.1 cusolver_12.1 cusolver_dev_12.1 cusparse_12.1 cusparse_dev_12.1 npp_12.1 npp_dev_12.1 nvjpeg_12.1 nvjpeg_dev_12.1 -n

0 commit comments

Comments
 (0)