From 583fed6c1742a4469c560fce75e025c37ade45ef Mon Sep 17 00:00:00 2001
From: Samuel Audet <samuel.audet@gmail.com>
Date: Mon, 8 Feb 2021 10:16:24 +0900
Subject: [PATCH] Add support for CUDA 11.2

Signed-off-by: Samuel Audet <samuel.audet@gmail.com>
---
 change-cuda-versions.sh | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/change-cuda-versions.sh b/change-cuda-versions.sh
index 634926d6cd0e..ee8081eb06bf 100755
--- a/change-cuda-versions.sh
+++ b/change-cuda-versions.sh
@@ -20,7 +20,7 @@
 
 set -e
 
-VALID_VERSIONS=( 9.2 10.0 10.1 10.2 11.0 11.1 )
+VALID_VERSIONS=( 9.2 10.0 10.1 10.2 11.0 11.1 11.2 )
 
 usage() {
   echo "Usage: $(basename $0) [-h|--help] <cuda version to be used>
@@ -47,6 +47,10 @@ check_cuda_version() {
 check_cuda_version "$VERSION"
 
 case $VERSION in
+  11.2)
+    VERSION2="8.1"
+    VERSION3="1.5.5-SNAPSHOT"
+    ;;
   11.1)
     VERSION2="8.0"
     VERSION3="1.5.5-SNAPSHOT"