From 23d8239cdf210e74ae57f7c6661c6237e5db6f75 Mon Sep 17 00:00:00 2001
From: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
Date: Mon, 22 Dec 2025 20:55:34 +0000
Subject: [PATCH] Change AMDGCN_USE_BUFFER_OPS default

---
 docker/Dockerfile.rocm | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm
index 1d84b6082fbd4..66869ba6316a5 100644
--- a/docker/Dockerfile.rocm
+++ b/docker/Dockerfile.rocm
@@ -145,6 +145,10 @@ ENV SAFETENSORS_FAST_GPU=1
 # Performance environment variable.
 ENV HIP_FORCE_DEV_KERNARG=1
 
+# Change the default to match upstream triton
+# https://github.com/triton-lang/triton/blob/main/python/triton/knobs.py#L505
+ENV AMDGCN_USE_BUFFER_OPS=1
+
 # Workaround for ROCm profiler limits
 RUN echo "ROCTRACER_MAX_EVENTS=10000000" > /app/libkineto.conf
 ENV KINETO_CONFIG="/app/libkineto.conf"