microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-04-19 22:39:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support CUDA 12.9 (#600)

Browse Source
This commit is contained in:
Changho Hwang
2025-08-08 12:02:50 -07:00
committed by GitHub
parent 9650e5c37e
commit 2c04b1be3d
4 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
.devcontainer/devcontainer.json
View File

@@ -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"
}
}
},

4
.devcontainer/devcontainer_amd.json
View File

@@ -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"
}
}
},

3
CMakeLists.txt
View File

@@ -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()

4
docker/build.sh
View File

@@ -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