From 4e5ab6e5471efe9acccc3d2290bc17d40196c473 Mon Sep 17 00:00:00 2001
From: Xiang Gao <qasdfgtyuiop@gmail.com>
Date: Wed, 29 Jan 2025 13:34:40 -0800
Subject: [PATCH] fix

---
 csrc/device_lower/pass/allocation.cpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/csrc/device_lower/pass/allocation.cpp b/csrc/device_lower/pass/allocation.cpp
index 0db9a146837..7c2924a3366 100644
--- a/csrc/device_lower/pass/allocation.cpp
+++ b/csrc/device_lower/pass/allocation.cpp
@@ -848,7 +848,11 @@ std::vector<Expr*> insertTMemRegionAllocsAndDeallocs(
 
     // the tcgen05.alloc instructions
     auto alloc_expr = IrBuilder::create<kir::AllocTMem>(
-        allocation_address, IrBuilder::create<Val>(0, DataType::UInt32));
+        allocation_address,
+        IrBuilder::create<Val>(
+            32,
+            DataType::UInt32) // TODO: hard code allocation size to 32 for now
+    );
     prologue.push_back(alloc_expr);
 
     // Relinquish the right to allocate after we are done with tcgen05.allocs