Add gfx90a target for ROCm CI (pytorch#106879)

...in preparation for upgrading CI runners to MI2xx Pull Request resolved: pytorch#106879 Approved by: https://github.com/seemethere
harborn · Aug 12, 2023 · aa36e16 · aa36e16
1 parent 6f83382
commit aa36e16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.ci/docker/build.sh b/.ci/docker/build.sh
@@ -375,7 +375,7 @@ docker build \
        --build-arg "NINJA_VERSION=${NINJA_VERSION:-}" \
        --build-arg "KATEX=${KATEX:-}" \
        --build-arg "ROCM_VERSION=${ROCM_VERSION:-}" \
-       --build-arg "PYTORCH_ROCM_ARCH=${PYTORCH_ROCM_ARCH:-gfx906}" \
+       --build-arg "PYTORCH_ROCM_ARCH=${PYTORCH_ROCM_ARCH:-gfx906;gfx90a}" \
        --build-arg "IMAGE_NAME=${IMAGE_NAME}" \
        --build-arg "UCX_COMMIT=${UCX_COMMIT}" \
        --build-arg "UCC_COMMIT=${UCC_COMMIT}" \