fix nvshmem cu126 (#9001)

48b8b4c1 · Yineng Zhang · GitHub · 323bc2f5 · 48b8b4c1
Unverified Commit 48b8b4c1 authored Aug 09, 2025 by Yineng Zhang Committed by GitHub Aug 09, 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
@@ -97,7 +97,7 @@ RUN cd /sgl-workspace/nvshmem && \
    NVSHMEM_PMIX_SUPPORT=0 \
    NVSHMEM_TIMEOUT_DEVICE_POLLING=0 \
    NVSHMEM_USE_GDRCOPY=1 \
-    cmake -S . -B build/ -DCMAKE_INSTALL_PREFIX=${NVSHMEM_DIR} -DCMAKE_CUDA_ARCHITECTURES="100;120" && \
+    cmake -S . -B build/ -DCMAKE_INSTALL_PREFIX=${NVSHMEM_DIR} -DCMAKE_CUDA_ARCHITECTURES="90" && \
    cmake --build build --target install -j${CMAKE_BUILD_PARALLEL_LEVEL}

 # Install DeepEP