From 3a6e9a595313af4c1c9b5eb6f816f9c25cc222a0 Mon Sep 17 00:00:00 2001
From: Bradley Dice <bdice@bradleydice.com>
Date: Mon, 22 Jul 2024 21:22:21 -0500
Subject: [PATCH] Encode conda CUDA support with a dictionary by release and
 major version.

---
 _includes/selector.html | 44 +++++++++++++++++++++++++++++------------
 1 file changed, 31 insertions(+), 13 deletions(-)

diff --git a/_includes/selector.html b/_includes/selector.html
index 209228e729f..c5efd91ab94 100644
--- a/_includes/selector.html
+++ b/_includes/selector.html
@@ -246,10 +246,10 @@
                 </div>
                 <div class="options-section" x-show="active_method === 'Conda'">
                     <div class="option-label">ENV. CUDA</div>
-                    <template x-for="version in conda_cuda_vers">
+                    <template x-for="version in conda_cuda_vers[active_release]">
                         <div x-on:click="(e) => condaCUDAClickHandler(e, version)"
                             x-bind:class="{'active': version === active_conda_cuda_ver, 'disabled': disableUnsupportedCuda(version)}"
-                            class="option" x-text="'CUDA ' + version"></div>
+                            class="option" x-text="'CUDA ' + conda_cuda_vers[active_release][version]['description']"></div>
                     </template>
                 </div>
                 <div class="options-section" x-show="active_method === 'pip'">
@@ -368,7 +368,7 @@
         Alpine.data('rapids_selector', () => ({
             // default values
             active_python_ver: "3.11",
-            active_conda_cuda_ver: "12.0 - 12.2",
+            active_conda_cuda_ver: "12",
             active_pip_cuda_ver: "12",
             active_docker_cuda_ver: "12.2",
             active_method: "Conda",
@@ -380,7 +380,31 @@
 
             // all possible values
             python_vers: ["3.9", "3.10", "3.11"],
-            conda_cuda_vers: ["11.4 - 11.8", "12.0 - 12.2", "12.0 - 12.5"],
+            // conda_cuda_vers is a dictionary structured such that
+            // conda_cuda_vers[release][version] gives a dictionary
+            // with a description and a cuda-version conda pinning.
+            conda_cuda_vers: {
+                "Stable": {
+                    "11": {
+                        "description": "11.4 - 11.8",
+                        "pinning": ">=11.4.*,<=11.8.*"
+                    },
+                    "12": {
+                        "description": "12.0 - 12.2",
+                        "pinning": ">=12.0.*,<=12.2.*"
+                    }
+                },
+                "Nightly": {
+                    "11": {
+                        "description": "11.4 - 11.8",
+                        "pinning": ">=11.4.*,<=11.8.*"
+                    },
+                    "12": {
+                        "description": "12.0 - 12.5",
+                        "pinning": ">=12.0.*,<=12.5.*"
+                    }
+                }
+            },
             pip_cuda_vers: ["11.4 - 11.8", "12"],
             docker_cuda_vers: ["11.8", "12.0", "12.2", "12.5"],
             methods: ["Conda", "pip", "Docker"],
@@ -447,8 +471,7 @@
                 var rapids_channel = this.active_release === "Stable" ? "rapidsai" : "rapidsai-nightly";
                 var dask_prefix = this.active_release === "Nightly" ? "dask/label/dev::" : "";
                 var python_version = this.active_python_ver;
-                var cuda_version_bounds = this.active_conda_cuda_ver.split(" - ");
-                var cuda_version_pinning = ">=" + cuda_version_bounds[0] + ".*,<=" + cuda_version_bounds[1] + ".*";
+                var cuda_version_pinning = this.conda_cuda_vers[this.active_release][this.active_conda_cuda_ver]["pinning"];
                 var py_cuda_pkgs = [this.highlightPkgOrImg("python") + "=" + python_version, this.highlightPkgOrImg("cuda-version") + cuda_version_pinning].join(" ");
                 var conda_channels = [rapids_channel, "conda-forge", "nvidia"]
                     .map(ch => "-" + this.highlightFlag("c") + " " + ch + " ")
@@ -512,7 +535,7 @@
                 var cuda_suffix = "-cu12";
                 var indentation = "    ";
 
-                if (this.active_pip_cuda_ver === "11.2 - 11.8") {
+                if (this.active_pip_cuda_ver === "11.4 - 11.8") {
                     cuda_suffix = "-cu11";
                 }
 
@@ -685,8 +708,6 @@
             disableUnsupportedCuda(cuda_version) {
                 var isDisabled = false;
                 if (this.active_additional_packages.includes("TensorFlow") && (!cuda_version.startsWith("12"))) isDisabled = true;
-                // Remove once stable releases support CUDA 12.5
-                if (this.active_release === "Stable" && cuda_version.includes("12.5")) isDisabled = true;
                 return isDisabled;
             },
             disableUnsupportedPython(python_version) {
@@ -717,9 +738,6 @@
                 if (this.isDisabled(e.target)) return;
                 this.active_release = release;
                 // Remove once stable releases support CUDA 12.5
-                if (this.active_release === "Stable" && this.active_conda_cuda_ver === "12.0 - 12.5") {
-                    this.active_conda_cuda_ver = "12.0 - 12.2";
-                }
                 if (this.active_release === "Stable" && this.active_docker_cuda_ver === "12.5") {
                     this.active_docker_cuda_ver = "12.2";
                 }
@@ -808,7 +826,7 @@
                     return;
                 }
                 this.active_additional_packages = [...this.active_additional_packages, package];
-                if (this.active_additional_packages.includes("TensorFlow") && (!this.active_conda_cuda_ver.includes("12.0"))) this.active_conda_cuda_ver = "12.0 - 12.2";
+                if (this.active_additional_packages.includes("TensorFlow") && (this.active_conda_cuda_ver !== "12")) this.active_conda_cuda_ver = "12";
             },
             copyToClipboard() {
                 let range = document.createRange();