* re-enable scheduleGlobalLoadLocalStore for Matmul kernels

triton-lang · Nov 15, 2024 · b789a7f · b789a7f
1 parent 7578863
commit b789a7f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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);
+      }
     }
   }
 };