Update ROCm to not hang when running on complex RNN models (#20118)

* Update ROCm to not hang when running on complex RNN models

* Formatting
This commit is contained in:
Nicolas Mowen
2025-09-17 19:26:32 -05:00
committed by GitHub
parent 1efff67e32
commit c05e260ae9
2 changed files with 12 additions and 0 deletions
+3
View File
@@ -67,6 +67,9 @@ FROM deps-prelim AS rocm-prelim-hsa-override0
ENV HSA_ENABLE_SDMA=0
ENV TF_ROCM_USE_IMMEDIATE_MODE=1
# avoid kernel crashes
ENV HIP_FORCE_DEV_KERNARG=1
COPY --from=rocm-dist / /
RUN ldconfig