configure build workflow for cuda12.8 and py3.11/3.13

Author: jringler-sidefx

.github/workflows/building.yml

@@ -29,56 +29,14 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-22.04, windows-2022]
-        python-version: ['3.10']
-        torch-version: ['2.0.0', '2.1.0', '2.2.0', '2.3.0', '2.4.0']
-        cuda-version: ['cu118', 'cu121', 'cu124']
+        python-version: ["3.11", "3.13"]
+        torch-version: ["2.9.1"]
+        cuda-version: ["cu128"]
         exclude:
-          - python-version: 3.12
-            torch-version: 2.0.0
-          - python-version: 3.12
-            torch-version: 2.1.0
-          - torch-version: 2.0.0
-            cuda-version: 'cu113'
-          - torch-version: 2.0.0
-            cuda-version: 'cu116'
-          - torch-version: 2.0.0
-            cuda-version: 'cu121'
-          - torch-version: 2.0.0
-            cuda-version: 'cu124'
-          - torch-version: 2.1.0
-            cuda-version: 'cu113'
-          - torch-version: 2.1.0
-            cuda-version: 'cu116'
-          - torch-version: 2.1.0
-            cuda-version: 'cu117'
-          - torch-version: 2.1.0
-            cuda-version: 'cu124'
-          - torch-version: 2.2.0
-            cuda-version: 'cu113'
-          - torch-version: 2.2.0
-            cuda-version: 'cu116'
-          - torch-version: 2.2.0
-            cuda-version: 'cu117'
-          - torch-version: 2.2.0
-            cuda-version: 'cu124'
-          - torch-version: 2.3.0
-            cuda-version: 'cu113'
-          - torch-version: 2.3.0
-            cuda-version: 'cu116'
-          - torch-version: 2.3.0
-            cuda-version: 'cu117'
-          - torch-version: 2.3.0
-            cuda-version: 'cu124'
-          - torch-version: 2.4.0
-            cuda-version: 'cu113'
-          - torch-version: 2.4.0
-            cuda-version: 'cu116'
-          - torch-version: 2.4.0
-            cuda-version: 'cu117'
-          - os: windows-2022
-            cuda-version: 'cu121'
-          - os: windows-2022
-            cuda-version: 'cu118'
+          - torch-version: "2.4.0"
+            cuda-version: "cu128"
+          - python-version: "3.13"
+            torch-version: "2.4.0"
 
     steps:
       - name: Checkout repository

.github/workflows/cuda/Linux-env.sh

@@ -3,6 +3,12 @@
 # Took from https://github.com/pyg-team/pyg-lib/
 
 case ${1} in
+  cu128)
+    export CUDA_HOME=/usr/local/cuda-12.8
+    export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
+    export PATH=${CUDA_HOME}/bin:${PATH}
+    export TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6;9.0;12.0"
+    ;;
   cu124)
     export CUDA_HOME=/usr/local/cuda-12.4
     export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
@@ -53,4 +59,4 @@ case ${1} in
     ;;
   *)
     ;;
-esac
+esac

.github/workflows/cuda/Linux.sh

@@ -5,6 +5,12 @@
 OS=ubuntu2004
 
 case ${1} in
+  cu128)
+    CUDA=12.8
+    APT_KEY=${OS}-${CUDA/./-}-local
+    FILENAME=cuda-repo-${APT_KEY}_${CUDA}.0-560.35.03-1_amd64.deb
+    URL=https://developer.download.nvidia.com/compute/cuda/${CUDA}.0/local_installers
+    ;;
   cu124)
     CUDA=12.4
     APT_KEY=${OS}-${CUDA/./-}-local
@@ -64,7 +70,7 @@ sudo mv cuda-${OS}.pin /etc/apt/preferences.d/cuda-repository-pin-600
 wget -nv ${URL}/${FILENAME}
 sudo dpkg -i ${FILENAME}
 
-if [ "${1}" = "cu117" ] || [ "${1}" = "cu118" ] || [ "${1}" = "cu121" ] || [ "${1}" = "cu124" ]; then
+if [ "${1}" = "cu117" ] || [ "${1}" = "cu118" ] || [ "${1}" = "cu121" ] || [ "${1}" = "cu124" ] || [ "${1}" = "cu128" ]; then
   sudo cp /var/cuda-repo-${APT_KEY}/cuda-*-keyring.gpg /usr/share/keyrings/
 else
   sudo apt-key add /var/cuda-repo-${APT_KEY}/7fa2af80.pub
@@ -74,4 +80,4 @@ sudo apt-get -qq update
 sudo apt install cuda-nvcc-${CUDA/./-} cuda-libraries-dev-${CUDA/./-} cuda-command-line-tools-${CUDA/./-}
 sudo apt clean
 
-rm -f ${FILENAME}
+rm -f ${FILENAME}

.github/workflows/cuda/Windows-env.sh

@@ -3,6 +3,12 @@
 # Took from https://github.com/pyg-team/pyg-lib/
 
 case ${1} in
+  cu128)
+    CUDA_HOME=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v12.8
+    PATH=${CUDA_HOME}/bin:$PATH
+    PATH=/c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/MSBuild/15.0/Bin:$PATH
+    export TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6;9.0;12.0"
+    ;;
   cu124)
     CUDA_HOME=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v12.4
     PATH=${CUDA_HOME}/bin:$PATH
@@ -47,4 +53,4 @@ case ${1} in
     ;;
   *)
     ;;
-esac
+esac

.github/workflows/cuda/Windows.sh

@@ -8,6 +8,11 @@ curl -k -L "https://drive.google.com/u/0/uc?id=1injUyo3lnarMgWyRcXqKg4UGnN0ysmuq
 7z x "/tmp/gpu_driver_dlls.zip" -o"/c/Windows/System32"
 
 case ${1} in
+  cu128)
+    CUDA_SHORT=12.8
+    CUDA_URL=https://developer.download.nvidia.com/compute/cuda/${CUDA_SHORT}.0/local_installers
+    CUDA_FILE=cuda_${CUDA_SHORT}.0_560.35.03_windows.exe
+    ;;
   cu124)
     CUDA_SHORT=12.4
     CUDA_URL=https://developer.download.nvidia.com/compute/cuda/${CUDA_SHORT}.1/local_installers
@@ -64,4 +69,4 @@ mkdir -p "/c/Program Files/NVIDIA Corporation/NvToolsExt/include"
 mkdir -p "/c/Program Files/NVIDIA Corporation/NvToolsExt/lib/x64"
 cp -r /tmp/NvToolsExt/bin/x64/* "/c/Program Files/NVIDIA Corporation/NvToolsExt/bin/x64"
 cp -r /tmp/NvToolsExt/include/* "/c/Program Files/NVIDIA Corporation/NvToolsExt/include"
-cp -r /tmp/NvToolsExt/lib/x64/* "/c/Program Files/NVIDIA Corporation/NvToolsExt/lib/x64"
+cp -r /tmp/NvToolsExt/lib/x64/* "/c/Program Files/NVIDIA Corporation/NvToolsExt/lib/x64"