Donato Capitella
51 rader
1.4 KiB
Bash
Executable File
51 rader
1.4 KiB
Bash
Executable File
#!/bin/bash
|
|
set -e
|
|
# Configuration
|
|
REPO_URL="https://github.com/kyuz0/rocm-systems.git"
|
|
BRANCH="gfx1151-rccl"
|
|
BUILD_DIR="build_gfx1151"
|
|
ROCM_PATH=${ROCM_PATH:-/opt/rocm}
|
|
# Project sub-directory
|
|
PROJECT_DIR="projects/rccl"
|
|
echo "=== Building RCCL for gfx1151 ==="
|
|
echo "Repo: $REPO_URL"
|
|
echo "Branch: $BRANCH"
|
|
echo "ROCm Path: $ROCM_PATH"
|
|
# 1. Clone/Fetch
|
|
if [ -d "rocm-systems" ]; then
|
|
echo "Directory 'rocm-systems' exists. Updating..."
|
|
cd rocm-systems
|
|
git fetch origin
|
|
git checkout $BRANCH
|
|
git pull origin $BRANCH
|
|
else
|
|
echo "Cloning repository..."
|
|
git clone -b $BRANCH $REPO_URL
|
|
cd rocm-systems
|
|
fi
|
|
# 2. Setup Build Directory
|
|
echo "Entering project directory..."
|
|
cd $PROJECT_DIR
|
|
mkdir -p $BUILD_DIR
|
|
cd $BUILD_DIR
|
|
echo "Configuring CMake for gfx1151..."
|
|
# We explicitly set GPU_TARGETS to gfx1151 to override the default list.
|
|
# We also set AMDGPU_TARGETS for standard rocm-cmake compliance.
|
|
CXX=$ROCM_PATH/bin/hipcc cmake .. \
|
|
-DCMAKE_CXX_COMPILER=$ROCM_PATH/bin/hipcc \
|
|
-DDEFAULT_GPUS="gfx1151" \
|
|
-DGPU_TARGETS="gfx1151" \
|
|
-DAMDGPU_TARGETS="gfx1151" \
|
|
-DCMAKE_INSTALL_PREFIX=./install \
|
|
-DBUILD_TESTS=OFF \
|
|
-DGENERATE_SYM_KERNELS=OFF \
|
|
-DENABLE_AMDSMI=OFF \
|
|
-DCMAKE_BUILD_TYPE=Release
|
|
# 3. Build
|
|
echo "Building librccl.so..."
|
|
make -j$(nproc)
|
|
echo "=== Build Complete ==="
|
|
echo "Libraries are located in:"
|
|
echo " $(pwd)/librccl.so"
|
|
echo " $(pwd)/librccl.so.1"
|