Skip to content

GitLab

Unverified Commit 48b8b4c1 authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

fix nvshmem cu126 (#9001)

parent 323bc2f5
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docker/Dockerfile
View file @ 48b8b4c1
...@@ -97,7 +97,7 @@ RUN cd /sgl-workspace/nvshmem && \ ...@@ -97,7 +97,7 @@ RUN cd /sgl-workspace/nvshmem && \
NVSHMEM_PMIX_SUPPORT=0 \ NVSHMEM_PMIX_SUPPORT=0 \
NVSHMEM_TIMEOUT_DEVICE_POLLING=0 \ NVSHMEM_TIMEOUT_DEVICE_POLLING=0 \
NVSHMEM_USE_GDRCOPY=1 \ 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} cmake --build build --target install -j${CMAKE_BUILD_PARALLEL_LEVEL}
# Install DeepEP # Install DeepEP
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment