[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:
co-authored by
Muhammad Ibrahim
Shengqi Chen
parent
494663940b
commit
6b24947628
+5
-1
@@ -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 \
|
||||
|
||||
Reference in New Issue
Block a user