Skip to content

Commit c74e084

Browse files
chuanqi129can-gaa-hou
authored andcommitted
[CD] Disable USE_MPI in XPU CI/CD wheel build (pytorch#159135)
XPU wheel build need source MPI for distributed XCCL backend build, but it also enable USE_MPI by default. Pull Request resolved: pytorch#159135 Approved by: https://github.com/malfet
1 parent 54ecc98 commit c74e084

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

.ci/manywheel/build_xpu.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ source /opt/intel/oneapi/mpi/latest/env/vars.sh
2525
export USE_STATIC_MKL=1
2626
export USE_ONEMKL=1
2727
export USE_XCCL=1
28+
export USE_MPI=0
2829

2930
WHEELHOUSE_DIR="wheelhousexpu"
3031
LIBTORCH_HOUSE_DIR="libtorch_housexpu"

.ci/pytorch/build.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -173,6 +173,7 @@ if [[ "$BUILD_ENVIRONMENT" == *xpu* ]]; then
173173
source /opt/intel/oneapi/mpi/latest/env/vars.sh
174174
# Enable XCCL build
175175
export USE_XCCL=1
176+
export USE_MPI=0
176177
# XPU kineto feature dependencies are not fully ready, disable kineto build as temp WA
177178
export USE_KINETO=0
178179
export TORCH_XPU_ARCH_LIST=pvc

0 commit comments

Comments
 (0)