[CI/Build] Build DeepEP for SM103 (B300/GB300) in the Docker image (#61829)

Signed-off-by: Muhammad Ibrahim <muhammad.ibrahim@multiversecomputing.com>
Co-authored-by: Muhammad Ibrahim <muhammad.ibrahim@multiversecomputing.com>
Co-authored-by: Shengqi Chen <harry-chen@outlook.com>
This commit is contained in:
Muhammad Ibrahim
2026-10-18 20:12:34 +08:00
committed by GitHub
co-authored by Muhammad Ibrahim Shengqi Chen
parent 494663940b
commit 6b24947628
+5 -1
View File
@@ -502,7 +502,11 @@ RUN --mount=type=cache,target=/opt/uv/cache \
> /tmp/nccl-override.txt && \
export UV_OVERRIDE=/tmp/nccl-override.txt; \
fi && \
export TORCH_CUDA_ARCH_LIST='9.0a 10.0a' && \
DEEPEP_ARCH_LIST='9.0a 10.0a' && \
if [ "$(printf '%s\n' 12.9 "$CUDA_VERSION" | sort -V | head -n1)" = 12.9 ]; then \
DEEPEP_ARCH_LIST="$DEEPEP_ARCH_LIST 10.3a"; \
fi && \
export TORCH_CUDA_ARCH_LIST="$DEEPEP_ARCH_LIST" && \
/tmp/install_python_libraries.sh \
--workspace /tmp/ep_kernels_workspace \
--mode wheel \