Skip to content

Commit 66c1286

Browse files
authored
Add GPU architecture env variables (#1250)
1 parent ec72ec8 commit 66c1286

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

.github/workflows/validate-domain-library.yml

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,9 @@ jobs:
9393
export PACKAGE_TYPE="${{ matrix.package_type }}"
9494
export CHANNEL="${{ matrix.channel }}"
9595
export SMOKE_TEST="${{ inputs.smoke_test }}"
96+
export GPU_ARCH_VER="${{ matrix.gpu_arch_version }}"
97+
export GPU_ARCH_TYPE="${{ matrix.gpu_arch_type }}"
98+
export DESIRED_CUDA="${{ matrix.desired_cuda }}"
9699
eval $SMOKE_TEST
97100
validate-windows:
98101
if: (inputs.os == 'windows' || inputs.os == 'all')
@@ -114,6 +117,9 @@ jobs:
114117
export PACKAGE_TYPE="${{ matrix.package_type }}"
115118
export CHANNEL="${{ matrix.channel }}"
116119
export SMOKE_TEST="${{ inputs.smoke_test }}"
120+
export GPU_ARCH_VER="${{ matrix.gpu_arch_version }}"
121+
export GPU_ARCH_TYPE="${{ matrix.gpu_arch_type }}"
122+
export DESIRED_CUDA="${{ matrix.desired_cuda }}"
117123
export TARGET_OS="windows"
118124
eval $SMOKE_TEST
119125
validate-macos:
@@ -136,6 +142,9 @@ jobs:
136142
export PACKAGE_TYPE="${{ matrix.package_type }}"
137143
export CHANNEL="${{ matrix.channel }}"
138144
export SMOKE_TEST="${{ inputs.smoke_test }}"
145+
export GPU_ARCH_VER="${{ matrix.gpu_arch_version }}"
146+
export GPU_ARCH_TYPE="${{ matrix.gpu_arch_type }}"
147+
export DESIRED_CUDA="${{ matrix.desired_cuda }}"
139148
eval $SMOKE_TEST
140149
validate-macos-arm64:
141150
if: (inputs.os == 'macos-arm64' || inputs.os == 'all')
@@ -157,4 +166,7 @@ jobs:
157166
export PACKAGE_TYPE="${{ matrix.package_type }}"
158167
export CHANNEL="${{ matrix.channel }}"
159168
export SMOKE_TEST="${{ inputs.smoke_test }}"
169+
export GPU_ARCH_VER="${{ matrix.gpu_arch_version }}"
170+
export GPU_ARCH_TYPE="${{ matrix.gpu_arch_type }}"
171+
export DESIRED_CUDA="${{ matrix.desired_cuda }}"
160172
eval $SMOKE_TEST

0 commit comments

Comments
 (0)