[Issue] JAX wheel build fails: `hip_blas_lt.cc references HIP_R_4F_E2M1_EXT missing from hipblaslt headers`

[chiranjeevipattigidi]

CI failure

Workflow: Release portable Linux JAX Wheels
Run: https://github.com/ROCm/TheRock/actions/runs/25366674695/

Context:

• Platform: Linux
• Impacted Archs: all the supported archs
• Jax versions: v0.8.2, v0.9.0

Run Links:

• https://github.com/ROCm/TheRock/actions/runs/25365352132/job/74374685167
• https://github.com/ROCm/TheRock/actions/runs/25364265225
• https://github.com/ROCm/TheRock/actions/runs/25363224052
• https://github.com/ROCm/TheRock/actions/runs/25361791608/job/74362845118
• https://github.com/ROCm/TheRock/actions/runs/25372889319/job/74400565077

Error

2026-05-05T11:38:35.1185877Z external/tsl/tsl/platform/protobuf.h:107:24: warning: 'uint8' is deprecated [-Wdeprecated-declarations]
2026-05-05T11:38:35.1186795Z   107 |       reinterpret_cast<uint8*>(output->data()));
2026-05-05T11:38:35.1187288Z       |                        ^
2026-05-05T11:38:35.1188003Z external/xla/xla/tsl/platform/types.h:33:13: note: 'uint8' has been explicitly marked deprecated here
2026-05-05T11:38:35.1188910Z    33 | using uint8 ABSL_DEPRECATE_AND_INLINE() = uint8_t;
2026-05-05T11:38:35.1189423Z       |             ^
2026-05-05T11:38:35.1190138Z external/com_google_absl/absl/base/macros.h:201:39: note: expanded from macro 'ABSL_DEPRECATE_AND_INLINE'
2026-05-05T11:38:35.1191269Z   201 | #define ABSL_DEPRECATE_AND_INLINE() [[deprecated, clang::annotate("inline-me")]]
2026-05-05T11:38:35.1191985Z       |                                       ^
2026-05-05T11:38:35.1193043Z external/xla/xla/stream_executor/rocm/hip_blas_lt.cc:628:23: error: use of undeclared identifier 'HIP_R_4F_E2M1_EXT'; did you mean 'HIP_R_8F_E5M3_EXT'?
2026-05-05T11:38:35.1194881Z   628 |   TYPED_MATMUL(float, HIP_R_4F_E2M1_EXT, HIP_R_4F_E2M1_EXT, HIP_R_32F, HIP_R_32F)
2026-05-05T11:38:35.1195585Z       |                       ^~~~~~~~~~~~~~~~~
2026-05-05T11:38:35.1196049Z       |                       HIP_R_8F_E5M3_EXT
2026-05-05T11:38:35.1196846Z external/xla/xla/stream_executor/rocm/hip_blas_lt.cc:555:35: note: expanded from macro 'TYPED_MATMUL'
2026-05-05T11:38:35.1197860Z   555 |   if (operand_types == std::tuple{ATYPE, BTYPE, CTYPE, DTYPE}) { \
2026-05-05T11:38:35.1198452Z       |                                   ^
2026-05-05T11:38:35.1199693Z bazel-out/k8-opt/bin/external/local_config_rocm/rocm/_virtual_includes/rocm_headers/rocm/include/hipblaslt/hipblaslt-types.h:78:11: note: 'HIP_R_8F_E5M3_EXT' declared here
2026-05-05T11:38:35.1201021Z    78 | int const HIP_R_8F_E5M3_EXT = 34;
2026-05-05T11:38:35.1201462Z       |           ^
2026-05-05T11:38:35.1202460Z external/xla/xla/stream_executor/rocm/hip_blas_lt.cc:628:42: error: use of undeclared identifier 'HIP_R_4F_E2M1_EXT'; did you mean 'HIP_R_8F_E5M3_EXT'?
2026-05-05T11:38:35.1203778Z   628 |   TYPED_MATMUL(float, HIP_R_4F_E2M1_EXT, HIP_R_4F_E2M1_EXT, HIP_R_32F, HIP_R_32F)
2026-05-05T11:38:35.1204496Z       |                                          ^~~~~~~~~~~~~~~~~
2026-05-05T11:38:35.1205009Z       |                                          HIP_R_8F_E5M3_EXT
2026-05-05T11:38:35.1206237Z external/xla/xla/stream_executor/rocm/hip_blas_lt.cc:555:42: note: expanded from macro 'TYPED_MATMUL'
2026-05-05T11:38:35.1207292Z   555 |   if (operand_types == std::tuple{ATYPE, BTYPE, CTYPE, DTYPE}) { \

[chiranjeevipattigidi]

failures seems related to hipblaslt and new hipblaslt changes introduced through latest rocm-libraries bump PR merged yesterday

changes: ROCm/rocm-libraries@2384d1b...89b3fc4

[WBobby]

Root cause: The rocm-libraries bump PR #5012 (2384d1b → 89b3fc4) pulled in hipBLASLt PR ROCm/rocm-libraries#6499 "[hipblaslt] Add support for gfx950 mxfp4", which deleted three _EXT constants from the public header projects/hipblaslt/library/include/hipblaslt/hipblaslt-types.h:

  - int const HIP_R_6F_E2M3_EXT = 31;
  - int const HIP_R_6F_E3M2_EXT = 32;
  - int const HIP_R_4F_E2M1_EXT = 33;
    int const HIP_R_8F_E5M3_EXT = 34;

@bnemanich Restore removed _EXT HIP datatype constants in hipblaslt-types.h