Support CUDA 12.9 (#600)

Author: chhwang

.devcontainer/devcontainer.json

@@ -3,7 +3,7 @@
     "build": {
         "dockerfile": "Dockerfile",
         "args": {
-            "BASE_IMAGE": "ghcr.io/microsoft/mscclpp/mscclpp:base-dev-cuda12.8",
+            "BASE_IMAGE": "ghcr.io/microsoft/mscclpp/mscclpp:base-dev-cuda12.9",
             "USERNAME": "devuser",
             "SSH_PORT": "22345"
         }
@@ -31,10 +31,8 @@
                     "/usr/local/cuda/include",
                     "/usr/include"
                 ],
-                "C_Cpp.default.compilerPath": "/usr/bin/gcc",
                 "C_Cpp.default.cStandard": "c17",
-                "C_Cpp.default.cppStandard": "c++17",
-                "C_Cpp.default.intelliSenseMode": "linux-gcc-x64"
+                "C_Cpp.default.cppStandard": "c++17"
             }
         }
     },

.devcontainer/devcontainer_amd.json

@@ -31,10 +31,8 @@
                     "/opt/rocm/include",
                     "/usr/include"
                 ],
-                "C_Cpp.default.compilerPath": "/usr/bin/gcc",
                 "C_Cpp.default.cStandard": "c17",
-                "C_Cpp.default.cppStandard": "c++17",
-                "C_Cpp.default.intelliSenseMode": "linux-gcc-x64"
+                "C_Cpp.default.cppStandard": "c++17"
             }
         }
     },

CMakeLists.txt

@@ -87,6 +87,9 @@ elseif(MSCCLPP_USE_CUDA)
     if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL "12.8")
         list(APPEND MSCCLPP_GPU_ARCHS 100)
     endif()
+    if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL "12.9")
+        list(APPEND MSCCLPP_GPU_ARCHS 120)
+    endif()
 elseif(MSCCLPP_USE_ROCM)
     set(CMAKE_HIP_ARCHITECTURES gfx90a gfx941 gfx942)
 endif()

docker/build.sh

@@ -10,6 +10,7 @@ baseImageTable=(
     ["cuda12.3"]="nvidia/cuda:12.3.2-devel-ubuntu20.04"
     ["cuda12.4"]="nvidia/cuda:12.4.1-devel-ubuntu22.04"
     ["cuda12.8"]="nvidia/cuda:12.8.1-devel-ubuntu22.04"
+    ["cuda12.9"]="nvidia/cuda:12.9.1-devel-ubuntu22.04"
     ["rocm6.2"]="rocm/rocm-terminal:6.2.1"
 )
 
@@ -25,13 +26,14 @@ declare -A ofedVersionTable
 ofedVersionTable=(
     ["cuda12.4"]="23.07-0.5.1.2"
     ["cuda12.8"]="24.10-1.1.4.0"
+    ["cuda12.9"]="24.10-1.1.4.0"
 )
 
 GHCR="ghcr.io/microsoft/mscclpp/mscclpp"
 TARGET=${1}
 
 print_usage() {
-    echo "Usage: $0 [cuda11.8|cuda12.1|cuda12.2|cuda12.3|cuda12.4|cuda12.8|rocm6.2]"
+    echo "Usage: $0 [cuda11.8|cuda12.1|cuda12.2|cuda12.3|cuda12.4|cuda12.8|cuda12.9|rocm6.2]"
 }
 
 if [[ ! -v "baseImageTable[${TARGET}]" ]]; then