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); + } } } };