MPICH 源码编译 with ucx with cuda，应用示例

Eloudy

已于 2024-09-16 01:30:03 修改

阅读量364

点赞数 2

文章标签： hpc hpl mpich

于 2024-09-16 01:27:58 首次发布

本文链接：https://blog.csdn.net/eloudy/article/details/142291518

版权

先基于 cuda 编译ucx

再基于 ucx 编译 mpich

mkdir mpich
mkdir ucx

1, 安装 ucx

预备环境：

sudo apt-get install valgrind
sudo apt-get install libibverbs-dev librdmacm-dev

下载ucx 源代码

git clone --recursive https://github.com/openucx/ucx.git
cd ucx/
git checkout v1.16.0

生成configure脚本：

./autogen.sh && ./autogen.sh
mkdir build
cd build/

配置：

../contrib/configure-devel --with-cuda=/usr/local/cuda  --without-rocm  --prefix=$PWD/../../local-dbg-cuda

release:
../contrib/configure-opt --with-cuda=/usr/local/cuda --prefix=$PWD/../../local-cuda

debug:
../contrib/configure-devel --with-cuda=/usr/local/cuda --without-rocm --prefix=$PWD/../../local-dbg-cuda

编译：


make -j
make install

2, 安装 mpich

下载源码：

 wget https://www.mpich.org/static/downloads/4.2.2/mpich-4.2.2.tar.gz
 tar zxf mpich-4.2.2.tar.gz

cd mpich-4.2.2/

配置：

./configure \
--prefix=/home/hipper/ex_mpich_2/local \
--enable-f90 \
--enable-cxx \
--enable-romio \
--enable-threads=multiple \
--with-ucx=/home/hipper/ex_mpich_2/ucx/local-dbg-cuda \
--with-ucx-lib=/home/hipper/ex_mpich_2/ucx/local-dbg-cuda/lib \
--with-ucx-include=/home/hipper/ex_mpich_2/ucx/local-dbg-cuda/include

结果：