[no-ci] toolshed: modernize conda_create_for_pathfinder_testing scripts

toolshed/conda_create_for_pathfinder_testing.ps1

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION
 # SPDX-License-Identifier: Apache-2.0
 
 param(
@@ -7,22 +7,30 @@ param(
 )
 
 $ErrorActionPreference = "Stop"
+Set-StrictMode -Version Latest
+
+$cudaMajor = $CudaVersion.Split(".", 2)[0]
+switch ($cudaMajor) {
+    "12" { $pythonVersion = "3.12" }
+    "13" { $pythonVersion = "3.14" }
+    default {
+        throw "Unsupported CUDA major version for this helper: $cudaMajor. Expected a 12.x or 13.x toolkit version."
+    }
+}
 
 & "$env:CONDA_EXE" "shell.powershell" "hook" | Out-String | Invoke-Expression
 
-conda create --yes -n "pathfinder_testing_cu$CudaVersion" python=3.13 "cuda-toolkit=$CudaVersion"
+conda create --yes -n "pathfinder_testing_cu$CudaVersion" "python=$pythonVersion" "cuda-toolkit=$CudaVersion"
 conda activate "pathfinder_testing_cu$CudaVersion"
 
+# Keep this list aligned with the Windows-installable subset of
+# cuda_pathfinder/pyproject.toml.
 $cpkgs = @(
     "cusparselt-dev",
     "cutensor",
-    "libcublasmp-dev",
+    "cutlass",
     "libcudss-dev",
-    "libcufftmp-dev",
-    "libmathdx-dev",
-    "libnvshmem3",
-    "libnvshmem-dev",
-    "libnvpl-fft-dev"
+    "libmathdx-dev"
 )
 
 foreach ($cpkg in $cpkgs) {

toolshed/conda_create_for_pathfinder_testing.sh

@@ -1,28 +1,65 @@
 #!/bin/bash
 
-# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+set -euo pipefail
+
 if [[ $# -ne 1 ]]; then
     echo "Usage: $(basename "$0") ctk-major-minor-patch" 1>&2
     exit 1
 fi
 
+cuda_version="$1"
+cuda_major="${cuda_version%%.*}"
+uname_m="$(uname -m)"
+case "$cuda_major" in
+    12)
+        python_version=3.12
+        ;;
+    13)
+        python_version=3.14
+        ;;
+    *)
+        echo "Unsupported CUDA major version for this helper: $cuda_major" 1>&2
+        echo "Expected a 12.x or 13.x toolkit version." 1>&2
+        exit 1
+        ;;
+esac
+
 eval "$(conda shell.bash hook)"
 
-conda create --yes -n "pathfinder_testing_cu$1" python=3.13 cuda-toolkit="$1"
-conda activate "pathfinder_testing_cu$1"
-
-for cpkg in \
-    cusparselt-dev \
-    cutensor \
-    libcublasmp-dev \
-    libcudss-dev \
-    libcufftmp-dev \
-    libmathdx-dev \
-    libnvshmem3 \
-    libnvshmem-dev \
-    libnvpl-fft-dev; do
+conda create --yes -n "pathfinder_testing_cu$cuda_version" "python=$python_version" cuda-toolkit="$cuda_version"
+set +u
+conda activate "pathfinder_testing_cu$cuda_version"
+set -u
+
+# Keep this list aligned with the Linux-installable subset of
+# cuda_pathfinder/pyproject.toml.
+cpkgs=(
+    "cusparselt-dev"
+    "cutensor"
+    "cutlass"
+    "libcublasmp-dev"
+    "libcudss-dev"
+    "libcufftmp-dev"
+    "libcusolvermp-dev"
+    "libmathdx-dev"
+    "libnvshmem3"
+    "libnvshmem-dev"
+)
+
+# Keep the conda environment aligned with platform-scoped pyproject groups.
+if [[ "$uname_m" == "aarch64" ]]; then
+    cpkgs+=("libnvpl-fft-dev")
+    if [[ "$cuda_major" == "13" ]]; then
+        cpkgs+=("libcudla-dev")
+    fi
+fi
+
+for cpkg in "${cpkgs[@]}"; do
     echo "CONDA INSTALL: $cpkg"
+    set +u
     conda install -y -c conda-forge "$cpkg"
+    set -u
 done