[AMD] Merge Dockerfiles for ROCm (#19203)

Co-authored-by: bingxche <Bingxu.Chen@amd.com>
This commit is contained in:
Alan Kao
2026-02-27 16:53:51 +08:00
committed by GitHub
parent 43fade5f69
commit 9b2fbf7e6a
6 changed files with 162 additions and 551 deletions

View File

@@ -190,13 +190,7 @@ elif [[ -n "${BUILD_FROM_DOCKERFILE}" ]]; then
fi
DOCKERFILE_DIR="${GITHUB_WORKSPACE:-$PWD}/docker"
# Use rocm720.Dockerfile for ROCm 7.2 builds, otherwise use rocm.Dockerfile
if [[ "${GPU_ARCH_BUILD}" == *"rocm720"* ]]; then
DOCKERFILE="${DOCKERFILE_DIR}/rocm720.Dockerfile"
else
DOCKERFILE="${DOCKERFILE_DIR}/rocm.Dockerfile"
fi
DOCKERFILE="${DOCKERFILE_DIR}/rocm.Dockerfile"
if [[ ! -f "${DOCKERFILE}" ]]; then
echo "Error: Dockerfile not found at ${DOCKERFILE}" >&2