From 02d9da36974fa3136bac875a735aace96d71f9c8 Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Thu, 5 Jun 2025 08:11:21 -0600
Subject: [PATCH] Add back in

---
 docker/rocm/Dockerfile | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docker/rocm/Dockerfile b/docker/rocm/Dockerfile
index f755c9f66..3bc28cae8 100644
--- a/docker/rocm/Dockerfile
+++ b/docker/rocm/Dockerfile
@@ -62,6 +62,7 @@ COPY --from=rocm /opt/rocm-dist/ /
 #######################################################################
 FROM deps-prelim AS rocm-prelim-hsa-override0
 ENV HSA_ENABLE_SDMA=0
+ENV TF_ROCM_USE_IMMEDIATE_MODE=1
 
 COPY --from=rocm-dist / /