Yiltan
e1ed36e58f
* Update install_dependencies.sh
* Updated to ROCm repos
* Merge pull request #37 from ROCm/depBuild
locked specific version on ompi and ucx
* locked specific version on ompi and ucx
* [IPC] Fix ROCSHMEM_SIGNAL_ADD
* Generate CMake Package Configuration Files
---------
Co-authored-by: akolliasAMD <99202231+akolliasAMD@users.noreply.github.com>
Co-authored-by: akolliasAMD <akollias@amd.com>
[ROCm/rocshmem commit: 785e31aa48]
73 lines
2.1 KiB
Bash
Executable File
73 lines
2.1 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -e
|
|
set -o pipefail
|
|
|
|
if [[ -z "${_ROCM_DIR}" ]]; then
|
|
export _ROCM_DIR=/opt/rocm
|
|
fi
|
|
|
|
# Location of dependencies source code
|
|
export _INSTALL_DIR=$BUILD_DIR/install
|
|
export _DEPS_SRC_DIR=$_INSTALL_DIR/src
|
|
|
|
mkdir -p $_DEPS_SRC_DIR
|
|
|
|
#Adjust branches and installation location as necessary
|
|
export _UCX_INSTALL_DIR=$_INSTALL_DIR/ucx
|
|
export _UCX_REPO=https://github.com/ROCm/ucx.git
|
|
export _UCX_COMMIT_HASH=4ef9a097c12ee6f7a8d3e41c317ea2d47e424b32
|
|
|
|
export _OMPI_INSTALL_DIR=$_INSTALL_DIR/ompi
|
|
export _OMPI_REPO=https://github.com/ROCm/ompi.git
|
|
export _OMPI_COMMIT_HASH=8a5c2ef25dc8e4528f0d3fd2ec91a6578160af95
|
|
|
|
# Step 1: Build UCX with ROCm support
|
|
cd $_DEPS_SRC_DIR
|
|
rm -rf ucx
|
|
git clone $_UCX_REPO
|
|
cd ucx
|
|
git checkout $_UCX_COMMIT_HASH
|
|
./autogen.sh
|
|
./contrib/configure-release --prefix=$_UCX_INSTALL_DIR \
|
|
--with-rocm=$_ROCM_DIR \
|
|
--enable-mt \
|
|
--without-go \
|
|
--without-java \
|
|
--without-cuda \
|
|
--without-verbs \
|
|
--without-knem
|
|
make -j
|
|
make install
|
|
|
|
# Step 2: Install OpenMPI with UCX support
|
|
cd $_DEPS_SRC_DIR
|
|
rm -rf ompi
|
|
git clone --recursive $_OMPI_REPO
|
|
cd ompi
|
|
git checkout $_OMPI_COMMIT_HASH
|
|
git submodule update --init --recursive
|
|
./autogen.pl
|
|
./configure --prefix=$_OMPI_INSTALL_DIR \
|
|
--with-rocm=$_ROCM_DIR \
|
|
--with-ucx=$_UCX_INSTALL_DIR \
|
|
--disable-oshmem \
|
|
--with-prrte=internal \
|
|
--with-hwloc=internal \
|
|
--with-libevent=internal \
|
|
--without-cuda \
|
|
--disable-sphinx \
|
|
--disable-mpi-fortran \
|
|
--without-ofi
|
|
make -j
|
|
make install
|
|
|
|
rm -rf $_DEPS_SRC_DIR
|
|
|
|
echo "Dependencies for rocSHMEM are now installed"
|
|
echo ""
|
|
echo "UCX ($_UCX_COMMIT_HASH) Installed to $_UCX_INSTALL_DIR"
|
|
echo "OpenMPI ($_OMPI_COMMIT_HASH) Installed to $_OMPI_INSTALL_DIR"
|
|
echo ""
|
|
echo "Please update your PATH and LD_LIBRARY_PATH"
|