From 3eaabdf8171574f7fccc4984d8e41dd63ee3d256 Mon Sep 17 00:00:00 2001
From: Xiang Gao <qasdfgtyuiop@gmail.com>
Date: Wed, 29 Mar 2023 11:52:33 -0700
Subject: [PATCH] Remove z-shape block swizzle

---
 csrc/scheduler/matmul.cpp | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/csrc/scheduler/matmul.cpp b/csrc/scheduler/matmul.cpp
index fcca8504846..660b4a291a7 100644
--- a/csrc/scheduler/matmul.cpp
+++ b/csrc/scheduler/matmul.cpp
@@ -181,7 +181,7 @@ void scheduleMatmul(
   // [... M,N,K]
   scheduler_utils::matmul_utils::makeTile(cc, gemm_tile.cta_tile.toVector());
 
-  // Applies swizzle factor on C
+  // Swizzle block tiles:
   if (params.grid_swizzle_factor != 1) {
     int factor = std::max(1, params.grid_swizzle_factor); // must be >=1
     if (params.rasterization_order ==
@@ -252,9 +252,6 @@ void scheduleMatmul(
   // ------------------------------------------------------------------
   // CTA tile:
 
-  // Swizzle block tiles:
-  c->swizzle(Swizzle2DType::ZShape, 0, 1, SwizzleMode::Loop);
-
   a->computeAt(c, 2);
   b->computeAt(c, 2);