Skip to content

Commit 960bbf4

Browse files
committed
Use vars.yml definition for picking CUDA architectures.
1 parent 75f1deb commit 960bbf4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/workflows/_build_torch_with_cuda.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ jobs:
2222
image: ${{ inputs.dev-image }}
2323
env:
2424
_GLIBCXX_USE_CXX11_ABI: 0
25-
TORCH_CUDA_ARCH_LIST: "5.2;8.6"
25+
TORCH_CUDA_ARCH_LIST: "7.0;7.5;8.0;9.0"
2626
USE_CUDA: 1
2727
MAX_JOBS: 24
2828
steps:

0 commit comments

Comments
 (0)