fix: shoudn't include CUDA_ARCH 100 and 120 for cuda12.6.1 (#11176)

0618ad6d · gongwei-130 · GitHub · 6a261aac · 0618ad6d
Unverified Commit 0618ad6d authored Oct 02, 2025 by gongwei-130 Committed by GitHub Oct 02, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

docker/Dockerfile docker/Dockerfile +1 -1

No files found.
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -109,7 +109,7 @@ RUN wget https://developer.download.nvidia.com/compute/redist/nvshmem/3.3.9/sour

 # Build and install NVSHMEM
 RUN cd /sgl-workspace/nvshmem && \
-    export CUDA_ARCH="90;100;120" && \
+    if [ "$BUILD_TYPE" = "blackwell" ] || [ "$BUILD_TYPE" = "blackwell_aarch" ]; then CUDA_ARCH="90;100;120"; else CUDA_ARCH="90"; fi && \
    NVSHMEM_SHMEM_SUPPORT=0 \
    NVSHMEM_UCX_SUPPORT=0 \
    NVSHMEM_USE_NCCL=0 \