Fabian
49 lines
1.5 KiB
Bash
49 lines
1.5 KiB
Bash
#!/bin/bash
|
|
# Build v4 - incremental rebuild after C10_WARP_SIZE fix
|
|
set -e
|
|
|
|
cd /home/fabian/pytorch-build
|
|
|
|
export HSA_OVERRIDE_GFX_VERSION=10.1.0
|
|
export HSA_ENABLE_SDMA=0
|
|
export HIP_VISIBLE_DEVICES=0
|
|
export ROCM_PATH=/opt/rocm
|
|
export HIP_PATH=/opt/rocm
|
|
export PYTORCH_ROCM_ARCH=gfx1010
|
|
export USE_ROCM=1
|
|
export USE_CUDA=0
|
|
export USE_FLASH_ATTENTION=0
|
|
export USE_MEM_EFF_ATTENTION=0
|
|
export USE_AOTRITON=0
|
|
export USE_TRITON=0
|
|
export MAX_JOBS=8
|
|
export CMAKE_POLICY_VERSION_MINIMUM=3.5
|
|
export CMAKE_PREFIX_PATH=/opt/rocm
|
|
export PYTHON_EXECUTABLE=/usr/bin/python3.11
|
|
|
|
LOG=/home/fabian/pytorch-build/build_v4.log
|
|
echo "============================================" | tee "$LOG"
|
|
echo " BUILD v4 STARTED at $(date)" | tee -a "$LOG"
|
|
echo " Incremental rebuild after C10_WARP_SIZE fix" | tee -a "$LOG"
|
|
echo "============================================" | tee -a "$LOG"
|
|
|
|
# Incremental build - ninja will only recompile changed files
|
|
cd build
|
|
cmake --build . --target install 2>&1 | tee -a "$LOG"
|
|
|
|
echo "============================================" | tee -a "$LOG"
|
|
echo " BUILD v4 FINISHED at $(date)" | tee -a "$LOG"
|
|
echo "============================================" | tee -a "$LOG"
|
|
|
|
# Back to source root for wheel
|
|
cd /home/fabian/pytorch-build
|
|
/usr/bin/python3.11 setup.py bdist_wheel 2>&1 | tee -a "$LOG"
|
|
|
|
WHL=$(ls dist/*.whl 2>/dev/null | head -1)
|
|
if [ -n "$WHL" ]; then
|
|
echo "SUCCESS: Wheel at $WHL" | tee -a "$LOG"
|
|
ls -lh "$WHL" | tee -a "$LOG"
|
|
else
|
|
echo "ERROR: No wheel produced!" | tee -a "$LOG"
|
|
fi
|