diff --git a/.ci/docker/almalinux/build.sh b/.ci/docker/almalinux/build.sh index 174ea7044954..ad234ce1ffb9 100755 --- a/.ci/docker/almalinux/build.sh +++ b/.ci/docker/almalinux/build.sh @@ -37,9 +37,9 @@ case ${DOCKER_TAG_PREFIX} in rocm*) BASE_TARGET=rocm PYTORCH_ROCM_ARCH="gfx900;gfx906;gfx908;gfx90a;gfx942;gfx1030;gfx1100;gfx1101;gfx1102;gfx1200;gfx1201" - # add gfx950 conditionally starting in ROCm 7.0 + # add gfx950, gfx115x conditionally starting in ROCm 7.0 if [[ "$ROCM_VERSION" == *"7.0"* ]]; then - PYTORCH_ROCM_ARCH="${PYTORCH_ROCM_ARCH};gfx950" + PYTORCH_ROCM_ARCH="${PYTORCH_ROCM_ARCH};gfx950;gfx1150;gfx1151" fi EXTRA_BUILD_ARGS="${EXTRA_BUILD_ARGS} --build-arg PYTORCH_ROCM_ARCH=${PYTORCH_ROCM_ARCH}" ;;