From bff9a1c2666de363d5b34f629e9fa61455a338ca Mon Sep 17 00:00:00 2001
From: Andreas Karatzas <akaratza@amd.com>
Date: Thu, 26 Mar 2026 13:33:45 -0500
Subject: [PATCH] [ROCm][CI] Override PYTORCH_ROCM_ARCH with detected GPU arch
 in test containers (#38165)

Signed-off-by: Andreas Karatzas <akaratza@amd.com>
---
 .buildkite/scripts/hardware_ci/run-amd-test.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.buildkite/scripts/hardware_ci/run-amd-test.sh b/.buildkite/scripts/hardware_ci/run-amd-test.sh
index e7dce35a1..472691d63 100755
--- a/.buildkite/scripts/hardware_ci/run-amd-test.sh
+++ b/.buildkite/scripts/hardware_ci/run-amd-test.sh
@@ -496,6 +496,7 @@ if is_multi_node "$commands"; then
 else
   echo "--- Single-node job"
   echo "Render devices: $BUILDKITE_AGENT_META_DATA_RENDER_DEVICES"
+
   docker run \
     --device /dev/kfd $BUILDKITE_AGENT_META_DATA_RENDER_DEVICES \
     $RDMA_FLAGS \
@@ -511,6 +512,7 @@ else
     -v "${HF_CACHE}:${HF_MOUNT}" \
     -e "HF_HOME=${HF_MOUNT}" \
     -e "PYTHONPATH=${MYPYTHONPATH}" \
+    -e "PYTORCH_ROCM_ARCH=" \
     --name "${container_name}" \
     "${image_name}" \
     /bin/bash -c "${commands}"