Skip to content

GitLab

Unverified Commit 0618ad6d authored by gongwei-130's avatar gongwei-130 Committed by GitHub
Browse files

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

parent 6a261aac
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docker/Dockerfile
View file @ 0618ad6d
...@@ -109,7 +109,7 @@ RUN wget https://developer.download.nvidia.com/compute/redist/nvshmem/3.3.9/sour ...@@ -109,7 +109,7 @@ RUN wget https://developer.download.nvidia.com/compute/redist/nvshmem/3.3.9/sour
# Build and install NVSHMEM # Build and install NVSHMEM
RUN cd /sgl-workspace/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_SHMEM_SUPPORT=0 \
NVSHMEM_UCX_SUPPORT=0 \ NVSHMEM_UCX_SUPPORT=0 \
NVSHMEM_USE_NCCL=0 \ NVSHMEM_USE_NCCL=0 \
......
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