Add CUDA 11.7 workflows (pytorch#81095)

Ref pytorch/builder#1042 CC @atalman Pull Request resolved: pytorch#81095 Approved by: https://github.com/atalman
thomasjpfan · Jul 18, 2022 · 99c464a · 99c464a
1 parent 35563f4
commit 99c464a
Show file tree

Hide file tree

Showing 10 changed files with 11,662 additions and 1,094 deletions.
diff --git a/.circleci/scripts/binary_linux_test.sh b/.circleci/scripts/binary_linux_test.sh
@@ -89,7 +89,7 @@ if [[ "$PACKAGE_TYPE" == conda ]]; then
 
       cu_ver="${DESIRED_CUDA:2:2}.${DESIRED_CUDA:4}"
       CUDA_PACKAGE="cudatoolkit"
-      if [[ "$DESIRED_CUDA" == "cu116" ]]; then
+      if [[ "$DESIRED_CUDA" == "cu116" || "$DESIRED_CUDA" == "cu117" ]]; then
         CUDA_PACKAGE="cuda"
       fi
 

diff --git a/.github/scripts/generate_binary_build_matrix.py b/.github/scripts/generate_binary_build_matrix.py
@@ -13,7 +13,7 @@
 from typing import Dict, List, Tuple, Optional
 
 
-CUDA_ARCHES = ["10.2", "11.3", "11.6"]
+CUDA_ARCHES = ["10.2", "11.3", "11.6", "11.7"]
 
 
 ROCM_ARCHES = ["5.0", "5.1.1"]