From 5e3da9b64e39a80ee83604ec580af3677cbd1f03 Mon Sep 17 00:00:00 2001
From: Lei Wang <34334180+LeiWang1999@users.noreply.github.com>
Date: Fri, 6 Sep 2024 20:18:19 +0800
Subject: [PATCH] [TL] Allow T.clear be applied on a "local" Buffer and improve
 L2 Swizzle (#178)

* Refactor BatchMatMulEmitter and BatchMatMulSelector for improved readability and maintainability

* Refactor import statements for improved readability and maintainability

* Refactor import statements for improved readability and maintainability

* disable failure email for ci

* remove email notifications.

* move relax pass from testing to mlc_llm

* Refactor scripts with se check_eual_ref_scripts_with_emitter function

* Lint Fix

* Refactor scripts with se check_eual_ref_scripts_with_emitter function

* buf fix for matrix support

* lint fix

* dispatch tensor core based on shapes

* update install commands

* import scripts

* remove shared mem hack

* revert change for swizzling

* bug fix

* tl examples

* Enhance Swizzle

* lint fix

* test fix

* lint fix

* optimize layout

* update tl utils.

* macro optimization

* test fix

* gemm_ss

* doc fix

* lint fix

* lint fix

* remove debug print

* remove debug print

* vectorization init

* lint fix
---
 3rdparty/tvm                         | 2 +-
 bitblas/base/roller/rasterization.py | 7 +++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/3rdparty/tvm b/3rdparty/tvm
index 39b2ba2fc..ba4a6ac13 160000
--- a/3rdparty/tvm
+++ b/3rdparty/tvm
@@ -1 +1 @@
-Subproject commit 39b2ba2fc24bf2ad441ef7b418c537c2814b21e2
+Subproject commit ba4a6ac132fd851cb55485c838e1df562082e1a7
diff --git a/bitblas/base/roller/rasterization.py b/bitblas/base/roller/rasterization.py
index 4fb779069..77afc1be2 100644
--- a/bitblas/base/roller/rasterization.py
+++ b/bitblas/base/roller/rasterization.py
@@ -7,12 +7,19 @@
 
 class Rasterization:
 
+    panel_width_ = None
+
     def __init__(self) -> None:
         pass
 
     def get_code(self) -> List[str]:
         raise NotImplementedError()
 
+    @property
+    def panel_width(self):
+        assert self.panel_width_ is not None
+        return self.panel_width_
+
 
 class NoRasterization(Rasterization):