Skip to content

Commit c1bb33c

Browse files
yyttt6yyttt6
committed
strip_decorators
1 parent 9aa92e1 commit c1bb33c

File tree

15 files changed

+1
-74
lines changed

15 files changed

+1
-74
lines changed

‎examples/attention_sink/bench_example_attention_sink.py‎

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -7,58 +7,44 @@
77
import example_mha_sink_fwd_bhsd_wgmma_pipelined
88

99

10-
@tilelang.testing.requires_cuda
1110
def bench_example_mha_sink_fwd_bhsd():
1211
tilelang.tools.bench.process_func(example_mha_sink_fwd_bhsd.main)
1312

1413

15-
@tilelang.testing.requires_cuda
1614
def bench_example_mha_sink_fwd_bhsd_sliding_window():
1715
tilelang.tools.bench.process_func(example_mha_sink_fwd_bhsd.main, window_size=128)
1816

1917

20-
@tilelang.testing.requires_cuda
21-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
2218
def bench_example_mha_sink_fwd_bhsd_wgmma_pipelined():
2319
tilelang.tools.bench.process_func(example_mha_sink_fwd_bhsd_wgmma_pipelined.main)
2420

2521

26-
@tilelang.testing.requires_cuda
27-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
2822
def bench_example_mha_sink_fwd_bhsd_wgmma_pipelined_sliding_window():
2923
tilelang.tools.bench.process_func(
3024
example_mha_sink_fwd_bhsd_wgmma_pipelined.main, window_size=128)
3125

3226

33-
@tilelang.testing.requires_cuda
34-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
3527
def bench_example_gqa_sink_fwd_bhsd_wgmma_pipelined():
3628
tilelang.tools.bench.process_func(example_gqa_sink_fwd_bhsd_wgmma_pipelined.main)
3729

3830

39-
@tilelang.testing.requires_cuda
40-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
4131
def bench_example_gqa_sink_fwd_bhsd_wgmma_pipelined_sliding_window():
4232
tilelang.tools.bench.process_func(
4333
example_gqa_sink_fwd_bhsd_wgmma_pipelined.main, window_size=128)
4434

4535

46-
@tilelang.testing.requires_cuda
4736
def bench_example_mha_sink_bwd_bhsd():
4837
tilelang.tools.bench.process_func(example_mha_sink_bwd_bhsd.main)
4938

5039

51-
@tilelang.testing.requires_cuda
5240
def bench_example_mha_sink_bwd_bhsd_sliding_window():
5341
tilelang.tools.bench.process_func(example_mha_sink_bwd_bhsd.main, window_size=128)
5442

5543

56-
@tilelang.testing.requires_cuda
5744
def bench_example_gqa_sink_bwd_bhsd():
5845
tilelang.tools.bench.process_func(example_gqa_sink_bwd_bhsd.main)
5946

6047

61-
@tilelang.testing.requires_cuda
6248
def bench_example_gqa_sink_bwd_bhsd_sliding_window():
6349
tilelang.tools.bench.process_func(example_gqa_sink_bwd_bhsd.main, window_size=128)
6450

‎examples/convolution/bench_example_convolution.py‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,9 @@
11
import tilelang.tools.bench
2+
import tilelang.testing
23
import example_convolution
34
import example_convolution_autotune
45

56

6-
@tilelang.testing.requires_cuda
7-
@tilelang.testing.requires_cuda_compute_version_le(8, 9)
87
def bench_example_convolution():
98
tilelang.tools.bench.process_func(example_convolution.main)
109

‎examples/deepseek_deepgemm/bench_example_deepgemm_fp8_2xAcc.py‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,6 @@
22
import example_deepgemm_fp8_2xAcc
33

44

5-
@tilelang.testing.requires_cuda
6-
@tilelang.testing.requires_cuda_compute_version_eq(9, 0)
75
def bench_example_deepgemm_fp8_2xAcc():
86
tilelang.tools.bench.process_func(example_deepgemm_fp8_2xAcc.main)
97

‎examples/deepseek_mla/bench_example_mla_decode.py‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,6 @@
22
import example_mla_decode
33

44

5-
@tilelang.testing.requires_cuda
6-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
75
def bench_example_mla_decode():
86
tilelang.tools.bench.process_func(example_mla_decode.main)
97

‎examples/deepseek_v32/bench_tilelang_example_deepseek_v32.py‎

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -14,20 +14,14 @@ def bench_fp8_lighting_indexer():
1414
tilelang.tools.bench.process_func(fp8_lighting_indexer.test_fp8_lighting_indexer)
1515

1616

17-
@tilelang.testing.requires_cuda
18-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
1917
def bench_sparse_mla_fwd():
2018
tilelang.tools.bench.process_func(sparse_mla_fwd.test_sparse_mla_fwd)
2119

2220

23-
@tilelang.testing.requires_cuda
24-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
2521
def bench_sparse_mla_fwd_pipelined():
2622
tilelang.tools.bench.process_func(sparse_mla_fwd_pipelined.test_sparse_mla_fwd_pipelined)
2723

2824

29-
@tilelang.testing.requires_cuda
30-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
3125
def bench_sparse_mla_bwd():
3226
tilelang.tools.bench.process_func(sparse_mla_bwd.test_sparse_mla_bwd)
3327

‎examples/dequantize_gemm/bench_example_dequantize_gemm.py‎

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -7,37 +7,26 @@
77
import example_dequant_groupedgemm_bf16_mxfp4_hopper
88

99

10-
@tilelang.testing.requires_cuda
1110
def bench_example_dequant_gemv_fp16xint4():
1211
tilelang.tools.bench.process_func(example_dequant_gemv_fp16xint4.main)
1312

1413

15-
@tilelang.testing.requires_cuda
16-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
1714
def bench_example_dequant_gemm_fp4_hopper():
1815
tilelang.tools.bench.process_func(example_dequant_gemm_fp4_hopper.main)
1916

2017

21-
@tilelang.testing.requires_cuda
22-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
2318
def bench_example_dequant_gemm_bf16_mxfp4_hopper():
2419
tilelang.tools.bench.process_func(example_dequant_gemm_bf16_mxfp4_hopper.main)
2520

2621

27-
@tilelang.testing.requires_cuda
28-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
2922
def bench_example_dequant_gemm_bf16_mxfp4_hopper_tma():
3023
tilelang.tools.bench.process_func(example_dequant_gemm_bf16_mxfp4_hopper_tma.main)
3124

3225

33-
@tilelang.testing.requires_cuda
34-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
3526
def bench_example_dequant_groupedgemm_bf16_mxfp4_hopper():
3627
tilelang.tools.bench.process_func(example_dequant_groupedgemm_bf16_mxfp4_hopper.main)
3728

3829

39-
@tilelang.testing.requires_cuda
40-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
4130
def bench_example_dequant_gemm_w4a8():
4231
tilelang.tools.bench.process_func(example_dequant_gemm_w4a8.main)
4332

‎examples/flash_attention/bench_example_flash_attention.py‎

Lines changed: 0 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -14,72 +14,54 @@
1414
import example_mha_fwd_varlen
1515

1616

17-
@tilelang.testing.requires_cuda
1817
def bench_example_gqa_bwd_tma_reduce_varlen():
1918
tilelang.tools.bench.process_func(example_gqa_bwd_tma_reduce_varlen.main)
2019

2120

22-
@tilelang.testing.requires_cuda
2321
def bench_example_gqa_bwd():
2422
tilelang.tools.bench.process_func(example_gqa_bwd.main)
2523

2624

27-
@tilelang.testing.requires_cuda
28-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
2925
def bench_example_gqa_bwd_wgmma_pipelined():
3026
tilelang.tools.bench.process_func(example_gqa_bwd_wgmma_pipelined.main)
3127

3228

33-
@tilelang.testing.requires_cuda
3429
def bench_example_mha_bwd():
3530
tilelang.tools.bench.process_func(example_mha_bwd.main)
3631

3732

38-
@tilelang.testing.requires_cuda
3933
def bench_example_mha_bwd_bhsd():
4034
tilelang.tools.bench.process_func(example_mha_bwd_bhsd.main)
4135

4236

43-
@tilelang.testing.requires_cuda
44-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
4537
def bench_example_mha_bwd_wgmma_pipelined():
4638
tilelang.tools.bench.process_func(example_mha_bwd_wgmma_pipelined.main)
4739

4840

49-
@tilelang.testing.requires_cuda
50-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
5141
def bench_example_gqa_fwd_bshd_wgmma_pipelined():
5242
tilelang.tools.bench.process_func(example_gqa_fwd_bshd_wgmma_pipelined.main)
5343

5444

55-
@tilelang.testing.requires_cuda
5645
def bench_example_gqa_fwd_bshd():
5746
tilelang.tools.bench.process_func(example_gqa_fwd_bshd.main)
5847

5948

60-
@tilelang.testing.requires_cuda
61-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
6249
def bench_example_mha_fwd_bhsd_wgmma_pipelined():
6350
tilelang.tools.bench.process_func(example_mha_fwd_bhsd_wgmma_pipelined.main)
6451

6552

66-
@tilelang.testing.requires_cuda
6753
def bench_example_mha_fwd_bhsd():
6854
tilelang.tools.bench.process_func(example_mha_fwd_bhsd.main)
6955

7056

71-
@tilelang.testing.requires_cuda
72-
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
7357
def bench_example_mha_fwd_bshd_wgmma_pipelined():
7458
tilelang.tools.bench.process_func(example_mha_fwd_bshd_wgmma_pipelined.main)
7559

7660

77-
@tilelang.testing.requires_cuda
7861
def bench_example_mha_fwd_bshd():
7962
tilelang.tools.bench.process_func(example_mha_fwd_bshd.main)
8063

8164

82-
@tilelang.testing.requires_cuda
8365
def bench_example_mha_fwd_varlen():
8466
tilelang.tools.bench.process_func(example_mha_fwd_varlen.main)
8567

‎examples/flash_decoding/bench_example_flash_decoding.py‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,6 @@
33
import example_mha_inference
44

55

6-
@tilelang.testing.requires_cuda
7-
@tilelang.testing.requires_cuda_compute_version_le(8, 9)
86
def bench_example_gqa_decode():
97
tilelang.tools.bench.process_func(example_gqa_decode.main)
108

‎examples/gemm_streamk/bench_example_tilelang_gemm_splitk.py‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,6 @@
22
import example_tilelang_gemm_streamk
33

44

5-
@tilelang.testing.requires_cuda
6-
@tilelang.testing.requires_cuda_compute_version_le(8, 9)
75
def bench_example_tilelang_gemm_streamk():
86
tilelang.tools.bench.process_func(example_tilelang_gemm_streamk.main)
97

‎examples/linear_attention/bench_linear_attn.py‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,10 @@
33
import example_linear_attn_fwd
44

55

6-
@tilelang.testing.requires_cuda
76
def bench_example_linear_attn_fwd():
87
tilelang.tools.bench.process_func(example_linear_attn_fwd.main)
98

109

11-
@tilelang.testing.requires_cuda
1210
def bench_example_linear_attn_bwd():
1311
tilelang.tools.bench.process_func(example_linear_attn_bwd.main)
1412

0 commit comments

Comments
 (0)