From b789a7fd516b57ebca3d19f8ae9fe472c7f9ee83 Mon Sep 17 00:00:00 2001
From: SJW <simon.waters@amd.com>
Date: Fri, 15 Nov 2024 23:12:23 +0000
Subject: [PATCH] * re-enable scheduleGlobalLoadLocalStore for Matmul kernels

---
 .../amd/lib/TritonAMDGPUTransforms/ReorderInstructions.cpp  | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/third_party/amd/lib/TritonAMDGPUTransforms/ReorderInstructions.cpp b/third_party/amd/lib/TritonAMDGPUTransforms/ReorderInstructions.cpp
index 85bcb8a0c7d0..abea5db773a2 100644
--- a/third_party/amd/lib/TritonAMDGPUTransforms/ReorderInstructions.cpp
+++ b/third_party/amd/lib/TritonAMDGPUTransforms/ReorderInstructions.cpp
@@ -366,10 +366,10 @@ struct TritonAMDGPUReorderInstructionsPass
 
       moveUpTranspose(funcOp);
 
-      if (isPureMatmulProblem(funcOp))
-        sinkSecondLoad(funcOp);
-      else
+      if (isPureMatmulProblem(funcOp)) {
         scheduleGlobalLoadLocalStore(funcOp);
+        sinkSecondLoad(funcOp);
+      }
     }
   }
 };