johnhe-dev
diff --git a/‎_includes/quick-start-module.js
+13-9 b/‎_includes/quick-start-module.js
+13-9
diff --git a/‎_includes/quick_start_local.html
+1-1 b/‎_includes/quick_start_local.html
+1-1
@@ -8,7 +8,7 @@ var supportedOperatingSystems = new Map([
 var supportedComputePlatforms = new Map([
   ['cuda10.2', new Set(['linux', 'windows'])],
   ['cuda11.x', new Set(['linux', 'windows'])],
-  ['cuda11.6', new Set(['linux', 'windows'])],
+  ['cuda11.y', new Set(['linux', 'windows'])],
   ['rocm5.x', new Set(['linux'])],
   ['accnone', new Set(['linux', 'macos', 'windows'])],
 ]);
@@ -117,14 +117,15 @@ function disableUnsupportedPlatforms(os) {
 
 // Change compute versions depending on build type
 function changeCUDAVersion(ptbuild) {
-  var cuda_element = document.getElementById("cuda11.x");
+  var cuda_element_x = document.getElementById("cuda11.x");
+  var cuda_element_y = document.getElementById("cuda11.y");
   var rocm_element = document.getElementById("rocm5.x");
-  if (cuda_element == null) {
-    console.log("Failed to find cuda11.x element");
+  if (cuda_element_x == null || cuda_element_y == null) {
+    console.log("Failed to find cuda11 elements");
     return;
   }
-  if (cuda_element.childElementCount != 1) {
-    console.log("Unexpected number of children for cuda11.x element");
+  if (cuda_element_x.childElementCount != 1 || cuda_element_y.childElementCount != 1) {
+    console.log("Unexpected number of children for cuda11 element");
     return;
   }
   if (rocm_element == null) {
@@ -137,13 +138,16 @@ function changeCUDAVersion(ptbuild) {
   }
   if (ptbuild == "preview") {
     rocm_element.children[0].textContent = "ROCm 5.2";
-    cuda_element.children[0].textContent = "CUDA 11.3";
+    cuda_element_x.children[0].textContent = "CUDA 11.6";
+    cuda_element_y.children[0].textContent = "CUDA 11.7";
   } else if (ptbuild == "stable") {
     rocm_element.children[0].textContent = "ROCm 5.1.1";
-    cuda_element.children[0].textContent = "CUDA 11.3";
+    cuda_element_x.children[0].textContent = "CUDA 11.3";
+    cuda_element_y.children[0].textContent = "CUDA 11.6";
   } else {
     rocm_element.children[0].textContent = "ROCm 5.1.1";
-    cuda_element.children[0].textContent = "CUDA 11.1";
+    cuda_element_x.children[0].textContent = "CUDA 11.3";
+    cuda_element_y.children[0].textContent = "CUDA 11.6";
   }
 }
 
 
@@ -94,7 +94,7 @@
       <div class="col-md-2 option block version" id="cuda11.x">
         <div class="option-text">CUDA 11.3</div>
       </div>
-      <div class="col-md-2 option block version" id="cuda11.6">
+      <div class="col-md-2 option block version" id="cuda11.y">
         <div class="option-text">CUDA 11.6</div>
       </div>
       <div class="col-md-3 option block version" id="rocm5.x">