Skip to content

Commit f7146af

Browse files
authored
[Typo] Remove disable_cache in some tests (tile-ai#755)
* Update test parameters and remove debug print statement - Adjusted test cases in `test_tilelang_dynamic_symbolic_bench.py` to use smaller matrix sizes (1024x1024) for improved performance and quicker execution. - Removed a debug print statement from `phase.py` to clean up the code and enhance clarity. * Refactor loop stack management in warp_specialized_rewriter - Introduced a new `LoopInfo` struct to encapsulate loop variable details, including `loop_var`, `extent`, and `min`, enhancing clarity and maintainability. - Updated the `loop_stack_` to utilize `LoopInfo` instead of a pair, improving type safety and readability. - Adjusted linear index calculations to account for the new structure, ensuring correct behavior in loop transformations. * Remove unused `torch.backends` import and `tilelang.disable_cache()` calls from multiple test files to enhance code clarity and maintainability.
1 parent 699335f commit f7146af

File tree

6 files changed

+1
-13
lines changed

6 files changed

+1
-13
lines changed

testing/python/cpu/test_tilelang_cpu_gemm.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,6 @@
44
import tilelang.language as T
55
import torch
66

7-
tilelang.disable_cache()
8-
97

108
def matmul(M, N, K, block_M, block_N, block_K, dtype="float16", accum_dtype="float"):
119
num_stages = 0

testing/python/dynamic/test_tilelang_dynamic_symbolic_bench.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,8 @@
11
import torch
2-
import torch.backends
32
from tilelang import tvm as tvm
43
import tilelang.testing
54
import tilelang.language as T
65

7-
tilelang.testing.set_random_seed(0)
8-
tilelang.disable_cache()
9-
106

117
def tl_matmul_block_static(
128
M,

testing/python/kernel/test_tilelang_kernel_int4_gemm_mma.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import torch
2-
import torch.backends
32
import tilelang
43
from tilelang import tvm as tvm
54
import tilelang.testing
@@ -14,7 +13,6 @@
1413
from tilelang.transform import simplify_prim_func
1514

1615
tilelang.testing.set_random_seed(42)
17-
tilelang.disable_cache()
1816

1917

2018
# @simplify_prim_func

testing/python/language/test_tilelang_language_annotate_pad.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,6 @@
33
import tilelang.testing
44
import torch
55

6-
tilelang.disable_cache()
7-
86

97
# add decorator @tilelang.jit if you want to return a torch function
108
# @tilelang.jit

testing/python/transform/test_tilelang_transform_config_index_bitwidth.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,6 @@
33
import tilelang
44
import tilelang.language as T
55

6-
tilelang.disable_cache()
7-
86

97
def blocksparse_flashattn(batch, heads, seq_len, dim, downsample_len, is_causal):
108
block_M = 64

testing/python/transform/test_tilelang_transform_thread_sync.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -189,4 +189,4 @@ def expected(A: T.Buffer((8192,), "float32")):
189189

190190

191191
if __name__ == "__main__":
192-
tilelang.disable_cache()
192+
tilelang.testing.main()

0 commit comments

Comments
 (0)