From bcfe641b26c802ea7d141a26a013b4cc6988353a Mon Sep 17 00:00:00 2001
From: Abdelrauf <rauf@konduit.ai>
Date: Tue, 9 Feb 2021 02:52:53 +0400
Subject: [PATCH] Add support for CUDA 11.2 (#9178)

Signed-off-by: Samuel Audet <samuel.audet@gmail.com>

Co-authored-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 719ca3fae..f402b01e4 100755
--- a/change-cuda-versions.sh
+++ b/change-cuda-versions.sh
@@ -26,7 +26,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>
@@ -53,6 +53,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"