Commit d6e8411
Make sure Windows CUDA 12.8 build follow same arches as Linux builds (pytorch#164477)
Make sure Windows CUDA 12.8 build follow same arches as Linux builds (pytorch#164470)
I believe ``set TORCH_CUDA_ARCH_LIST=7.0;7.5;8.0;8.6;9.0;10.0;12.0`` is the one thats actually used. Hence remove 6.1 to align the support with Linux support.
Pull Request resolved: pytorch#164470
Approved by: https://github.com/tinglvv, https://github.com/nWEIdia, https://github.com/Camyll
(cherry picked from commit 235b995)
Co-authored-by: Andrey Talman <[email protected]>1 parent 10b501f commit d6e8411
1 file changed
+2
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
37 | 37 | | |
38 | 38 | | |
39 | 39 | | |
40 | | - | |
| 40 | + | |
41 | 41 | | |
42 | 42 | | |
43 | | - | |
| 43 | + | |
44 | 44 | | |
45 | 45 | | |
46 | 46 | | |
| |||
0 commit comments