bug fix

Author: LeiWang1999

tilelang/language/utils.py

@@ -54,8 +54,7 @@ def buffer_load_to_tile_region(load: BufferLoad, access_type: str, extents: List
         tir.Call: A region descriptor for the loaded area
     """
     indices = load.indices
-    print("indices", indices)
-    print("extents", extents)
+
     if len(indices) > len(extents):
         # (f"mismatch between indices and extents for buffer load {load}: indices = {indices}, extents = {extents}, "
         # f"region will be expanded in the last 2 dimensions")

tilelang/utils/language.py

@@ -131,11 +131,16 @@ def get_buffer_region_from_load(buffer_load: tir.BufferLoad) -> Optional[tir.Buf
     """
     buffer, indices = buffer_load.buffer, buffer_load.indices
     regions = []
+    found_ramp: bool = False
     for indice in indices:
         if isinstance(indice, tir.Ramp):
             regions.append(ir.Range.from_min_extent(indice.base, indice.lanes))
+            found_ramp = True
         elif isinstance(indice, tir.PrimExpr):
             regions.append(ir.Range.from_min_extent(indice, 1))
         else:
             raise ValueError("Unsupported type: ", type(indice))
-    return tir.BufferRegion(buffer, regions)
+    if found_ramp:
+        return tir.BufferRegion(buffer, regions)
+    else:
+        return None