From 89401baa8bfdf3fe04c83ee5fac3e3f20431d3db Mon Sep 17 00:00:00 2001 From: Xing-lil Date: Mon, 17 Mar 2025 12:01:41 +0800 Subject: [PATCH 1/2] dockerfile add flagcx --- tools/dockerfile/ci_dockerfile.sh | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tools/dockerfile/ci_dockerfile.sh b/tools/dockerfile/ci_dockerfile.sh index 27414995235be..1e1ad88678ca4 100644 --- a/tools/dockerfile/ci_dockerfile.sh +++ b/tools/dockerfile/ci_dockerfile.sh @@ -126,6 +126,10 @@ function make_ubuntu20_cu123_dockerfile(){ sed -i 's#RUN bash /build_scripts/install_cudnn.sh cudnn841#RUN bash /build_scripts/install_cudnn.sh cudnn900 #g' ${dockerfile_name} sed -i 's#CUDNN_VERSION=8.4.1#CUDNN_VERSION=9.0.0#g' ${dockerfile_name} + sed -i "${dockerfile_line}i RUN git clone https://github.com/FlagOpen/FlagCX.git /FlagCX \&\& \ + cd /FlagCX \&\& \ + make USE_NVIDIA=1 CCL_HOME=/usr/local/cuda-12.3/targets/x86_64-linux/lib" ${dockerfile_name} + sed -i "${dockerfile_line}i WORKDIR /home \n \ RUN git clone --depth=1 https://github.com/PaddlePaddle/PaddleNLP.git -b stable/paddle-ci \&\& cd PaddleNLP \&\& \ pip3.10 install -r requirements.txt \&\& \ From 852cc00ca91830dfae2b4deedd1f389f75f17399 Mon Sep 17 00:00:00 2001 From: ZhenxingLi Date: Mon, 17 Mar 2025 12:17:52 +0800 Subject: [PATCH 2/2] Update CMakeLists.txt --- CMakeLists.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/CMakeLists.txt b/CMakeLists.txt index a36c8183457a0..04266f492b2b2 100755 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -312,6 +312,7 @@ option( "Choose the type of sanitizer, options are: Address, Leak, Memory, Thread, Undefined" OFF) option(WITH_CINN "Compile PaddlePaddle with CINN" OFF) +option(WITH_FLAGCX "Compile PaddlePaddle with FLAGCX support" ON) option(WITH_NCCL "Compile PaddlePaddle with NCCL support" ON) option(WITH_RCCL "Compile PaddlePaddle with RCCL support" ON) option(WITH_XPU_BKCL "Compile PaddlePaddle with BAIDU KUNLUN XPU BKCL" OFF)