From 0d1f3db28016fe52d2af5df7308c9bc1eac4a2b7 Mon Sep 17 00:00:00 2001
From: pbialecki <pbialecki@nvidia.com>
Date: Wed, 8 Jun 2022 00:21:01 -0700
Subject: [PATCH 1/2] add CUDA 11.7 builds for manywheel and libtorch

---
 .github/workflows/build-libtorch-images.yml  | 2 +-
 .github/workflows/build-manywheel-images.yml | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/build-libtorch-images.yml b/.github/workflows/build-libtorch-images.yml
index 742ca00c7..cf24caae6 100644
--- a/.github/workflows/build-libtorch-images.yml
+++ b/.github/workflows/build-libtorch-images.yml
@@ -28,7 +28,7 @@ jobs:
     runs-on: ubuntu-18.04
     strategy:
       matrix:
-        cuda_version: ["11.6", "11.5", "11.3", "10.2"]
+        cuda_version: ["11.7", "11.6", "11.5", "11.3", "10.2"]
     env:
       GPU_ARCH_TYPE: cuda
       GPU_ARCH_VERSION: ${{ matrix.cuda_version }}
diff --git a/.github/workflows/build-manywheel-images.yml b/.github/workflows/build-manywheel-images.yml
index ce5cc12d0..f2d6afb8d 100644
--- a/.github/workflows/build-manywheel-images.yml
+++ b/.github/workflows/build-manywheel-images.yml
@@ -28,7 +28,7 @@ jobs:
     runs-on: ubuntu-18.04
     strategy:
       matrix:
-        cuda_version: ["11.6", "11.5", "11.3", "10.2"]
+        cuda_version: ["11.7", "11.6", "11.5", "11.3", "10.2"]
     env:
       GPU_ARCH_TYPE: cuda
       GPU_ARCH_VERSION: ${{ matrix.cuda_version }}

From b0983e629ff2550dffb052342fcb1ef649d057f5 Mon Sep 17 00:00:00 2001
From: pbialecki <pbialecki@nvidia.com>
Date: Thu, 9 Jun 2022 20:09:53 -0700
Subject: [PATCH 2/2] add libtorch and magma builds

---
 common/install_magma.sh      | 3 +++
 libtorch/Dockerfile          | 4 ++++
 libtorch/build_all_docker.sh | 2 +-
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/common/install_magma.sh b/common/install_magma.sh
index ea2548140..5d14dbfe2 100644
--- a/common/install_magma.sh
+++ b/common/install_magma.sh
@@ -11,6 +11,9 @@ function do_install() {
     if [[ ${cuda_version_nodot} == 116 ]]; then
         MAGMA_VERSION="2.6.1"
         magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-0.tar.bz2"
+    elif [[ ${cuda_version_nodot} == 117 ]]; then
+	MAGMA_VERSION="2.6.1"
+	magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-0.tar.bz2"
     else
         magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-1.tar.bz2"
     fi
diff --git a/libtorch/Dockerfile b/libtorch/Dockerfile
index d5b22b09e..8ae734d2d 100644
--- a/libtorch/Dockerfile
+++ b/libtorch/Dockerfile
@@ -60,6 +60,10 @@ FROM cuda as cuda11.6
 RUN bash ./install_cuda.sh 11.6
 RUN bash ./install_magma.sh 11.6
 
+FROM cuda as cuda11.7
+RUN bash ./install_cuda.sh 11.7
+RUN bash ./install_magma.sh 11.7
+
 FROM cpu as rocm
 ARG PYTORCH_ROCM_ARCH
 ENV PYTORCH_ROCM_ARCH ${PYTORCH_ROCM_ARCH}
diff --git a/libtorch/build_all_docker.sh b/libtorch/build_all_docker.sh
index 1493778e4..208242805 100755
--- a/libtorch/build_all_docker.sh
+++ b/libtorch/build_all_docker.sh
@@ -4,7 +4,7 @@ set -eou pipefail
 
 TOPDIR=$(git rev-parse --show-toplevel)
 
-for cuda_version in 11.6 11.5 11.3 10.2; do
+for cuda_version in 11.7 11.6 11.5 11.3 10.2; do
     GPU_ARCH_TYPE=cuda GPU_ARCH_VERSION="${cuda_version}" "${TOPDIR}/libtorch/build_docker.sh"
 done