60 řádky
1.7 KiB
Bash
Spustitelný soubor
60 řádky
1.7 KiB
Bash
Spustitelný soubor
#!/bin/bash
|
|
set -e
|
|
# Configuration
|
|
REPO_URL="https://code.badstorm.xyz/AI/rocm-systems.git"
|
|
BRANCH="gfx1150-rccl"
|
|
BUILD_DIR="build_gfx1150"
|
|
ROCM_PATH=${ROCM_PATH:-/opt/rocm}
|
|
# Project sub-directory
|
|
PROJECT_DIR="projects/rccl"
|
|
echo "=== Building RCCL for gfx1150 ==="
|
|
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 gfx1150..."
|
|
# We explicitly set GPU_TARGETS to gfx1150 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="gfx1150" \
|
|
-DGPU_TARGETS="gfx1150" \
|
|
-DAMDGPU_TARGETS="gfx1150" \
|
|
-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)
|
|
|
|
# Comprimi il file reale (non il symlink)
|
|
cd /home/badstorm/Source/ai/amd-strix-halo-vllm-toolboxes
|
|
gzip -k rocm-systems/projects/rccl/build_gfx1150/librccl.so.1.0
|
|
|
|
# Copia i file .gz in custom_libs/
|
|
mkdir -p custom_libs/
|
|
cp rocm-systems/projects/rccl/build_gfx1150/librccl.so.1.0.gz custom_libs/librccl.so.1.gz
|
|
|
|
echo "=== Build Complete ==="
|
|
echo "Libraries are located in:"
|
|
echo " $(pwd)/librccl.so"
|
|
echo " $(pwd)/librccl.so.1"
|