Skip to content

Commit 8404fda

Browse files
committed
Fix typos in rasterization parameters and update import path for cached module
- Corrected the spelling of `enable_rasteration` to `enable_rasterization` in the matmul function and its usage. - Updated the import statement for the `cached` module to reflect the new path in the cache submodule. - Added `StridedTensor` import in the language module for enhanced tensor functionality.
1 parent e5dd402 commit 8404fda

File tree

3 files changed

+4
-3
lines changed

3 files changed

+4
-3
lines changed

testing/python/autotune/test_tilelang_autotune_with_inputs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def matmul(M,
5252
block_K=32,
5353
num_stages=0,
5454
thread_num=128,
55-
enable_rasteration=False):
55+
enable_rasterization=False):
5656

5757
dtype = "float16"
5858
accum_dtype = "float"
@@ -85,7 +85,7 @@ def main(
8585
C_local = T.alloc_fragment((block_M, block_N), accum_dtype)
8686

8787
# Enable (or disable) swizzling optimization
88-
T.use_swizzle(panel_size=10, enable=enable_rasteration)
88+
T.use_swizzle(panel_size=10, enable=enable_rasterization)
8989

9090
# Clear out the accumulation buffer
9191
T.clear(C_local)

testing/python/cache/test_tilelang_cache_matmul.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from tilelang import tvm as tvm
22
import tilelang.testing
3-
from tilelang import cached
3+
from tilelang.cache import cached
44
import tilelang.language as T
55

66

tilelang/language/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
make_tensor, # noqa: F401
1818
Buffer, # noqa: F401
1919
Tensor, # noqa: F401
20+
StridedTensor, # noqa: F401
2021
FragmentBuffer, # noqa: F401
2122
SharedBuffer, # noqa: F401
2223
LocalBuffer, # noqa: F401

0 commit comments

Comments
 (0)