[AMD][CI] Swap the AMD PR gate to ROCm 7.2 and demote ROCm 7.0 to a daily shadow (#34204)
Co-authored-by: yctseng0211 <yctseng@smci355-ccs-aus-m12-17.cs-aus.dcgpu> Co-authored-by: Chen <bingxche@amd.com>
This commit is contained in:
co-authored by
yctseng0211
Chen
parent
b3bffef70a
commit
00bdafe944
@@ -238,6 +238,19 @@ DOCKERFILE="docker/rocm.Dockerfile"
|
||||
GPU_ARCH="${GPU_ARCH:-mi30x}"
|
||||
echo "[CI-AITER-CHECK] Runner GPU_ARCH=${GPU_ARCH}"
|
||||
|
||||
# ROCm 7.0 keeps the Triton its base image ships; later ROCm images run on the
|
||||
# Triton AITER pins, so a rebuilt AITER has to bring its own along.
|
||||
IMAGE_HIP_VERSION=$(docker exec ci_sglang python3 -c 'import torch; print(torch.version.hip or "")')
|
||||
case "${IMAGE_HIP_VERSION}" in
|
||||
7.0*) INSTALL_AITER_TRITON="false" ;;
|
||||
7.*) INSTALL_AITER_TRITON="true" ;;
|
||||
*)
|
||||
echo "[CI-AITER-CHECK] ERROR: Unsupported or empty HIP version: '${IMAGE_HIP_VERSION}'"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
echo "[CI-AITER-CHECK] Container HIP=${IMAGE_HIP_VERSION}, install AITER's Triton on rebuild=${INSTALL_AITER_TRITON}"
|
||||
|
||||
#############################################
|
||||
# 1. Extract AITER_COMMIT from correct Dockerfile block
|
||||
#############################################
|
||||
@@ -292,7 +305,6 @@ else
|
||||
NEED_REBUILD="true"
|
||||
fi
|
||||
|
||||
|
||||
#############################################
|
||||
# 4. Rebuild AITER if needed
|
||||
#############################################
|
||||
@@ -329,6 +341,18 @@ if [[ "${NEED_REBUILD}" == "true" ]]; then
|
||||
fi
|
||||
echo "[CI-AITER-CHECK] GPU_ARCH_LIST=${GPU_ARCH_LIST}"
|
||||
|
||||
# Run the installer here rather than letting setup.py do it: setup.py
|
||||
# swallows its errors, and the AITER_USE_SYSTEM_TRITON=1 below then keeps
|
||||
# whatever Triton is already installed. Doing it up front fails closed.
|
||||
if [[ "${INSTALL_AITER_TRITON}" == "true" ]]; then
|
||||
docker exec ci_sglang bash -c "
|
||||
set -euo pipefail
|
||||
cd /sgl-workspace/aiter
|
||||
test -f .github/scripts/install_triton.sh
|
||||
bash .github/scripts/install_triton.sh
|
||||
"
|
||||
fi
|
||||
|
||||
# build AITER
|
||||
docker exec ci_sglang bash -c "
|
||||
cd /sgl-workspace/aiter && \
|
||||
|
||||
Reference in New Issue
Block a user