From 6fe1f1cae8f9b4fdc52934a3656eaa69685c8668 Mon Sep 17 00:00:00 2001
From: tsrw2048 <239799652+tsrw2048@users.noreply.github.com>
Date: Wed, 20 May 2026 19:47:08 +0000
Subject: [PATCH] Bundle specific and generic arch targets together.

This is needed in order to build release wheels with generic targets
while also passing all unit tests. ROCPrim currently has an issue with
building cub sort for generic arch targets so the specific arch target
should be used here instead.
---
 build/rocm/rocm.bazelrc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/build/rocm/rocm.bazelrc b/build/rocm/rocm.bazelrc
index f656d2f6764e..0e0b3d07a0ae 100644
--- a/build/rocm/rocm.bazelrc
+++ b/build/rocm/rocm.bazelrc
@@ -17,7 +17,7 @@ common:rocm --action_env=CLANG_COMPILER_PATH="/usr/lib/llvm-18/bin/clang"
 common:rocm --action_env=HIPCC_COMPILE_FLAGS_APPEND=--offload-compress
 common:rocm --copt=-Wno-gnu-offsetof-extensions
 common:rocm --copt=-Qunused-arguments
-common:rocm --repo_env=TF_ROCM_AMDGPU_TARGETS="gfx908,gfx90a,gfx942,gfx950,gfx1030,gfx1100,gfx1101,gfx1200,gfx1201"
+common:rocm --repo_env=TF_ROCM_AMDGPU_TARGETS="gfx908,gfx90a,gfx942,gfx950,gfx1030,gfx1100,gfx1101,gfx1200,gfx1201,gfx9-4-generic,gfx10-3-generic,gfx11-generic,gfx12-generic"
 
 # Used for @xla//build_tools/rocm:parallel_gpu_execute
 common:rocm --legacy_external_runfiles=true