---
id: pr-tilelang-1568
repo: tile-ai/tilelang
pr: 1568
title: '[Refactor] Simplify do_bench calls by using default warmup and rep parameters'
author: LeiWang1999
date: '2025-12-29'
url: https://github.com/tile-ai/tilelang/pull/1568
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- attention kernel
- moe
- moe kernel
techniques: []
hardware_features: []
kernel_types:
- attention
- moe
languages:
- cuda-cpp
captured_at: '2026-05-16'
status: merged
merge_sha: e23bce79d68095b5ccd60fd53bf8029891ac3166
inclusion_reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.
changed_paths:
- examples/attention_sink/example_gqa_sink_bwd_bhsd.py
- examples/attention_sink/example_gqa_sink_fwd_bhsd_wgmma_pipelined.py
- examples/attention_sink/example_mha_sink_bwd_bhsd.py
- examples/attention_sink/example_mha_sink_fwd_bhsd.py
- examples/attention_sink/example_mha_sink_fwd_bhsd_wgmma_pipelined.py
- examples/blocksparse_attention/example_tilelang_block_sparse_attn.py
- examples/blocksparse_attention/example_tilelang_sparse_gqa_decode_paged.py
- examples/blocksparse_attention/example_tilelang_sparse_gqa_decode_varlen_indice.py
- examples/blocksparse_attention/example_tilelang_sparse_gqa_decode_varlen_mask.py
- examples/blocksparse_gemm/example_blocksparse_gemm.py
- examples/cast/example_per_token_cast_to_fp8.py
- examples/deepseek_mla/example_mla_decode.py
- examples/deepseek_nsa/example_tilelang_nsa_decode.py
- examples/deepseek_nsa/example_tilelang_nsa_fwd.py
- examples/deepseek_v32/fp8_lighting_indexer.py
- examples/deepseek_v32/sparse_mla_bwd.py
- examples/deepseek_v32/sparse_mla_fwd.py
- examples/deepseek_v32/sparse_mla_fwd_pipelined.py
- examples/deepseek_v32/topk_selector.py
- examples/dequantize_gemm/example_dequant_gemm_bf16_fp4_hopper.py
- examples/dequantize_gemm/example_dequant_gemm_bf16_mxfp4_hopper.py
- examples/dequantize_gemm/example_dequant_gemm_fp4_hopper.py
- examples/dequantize_gemm/example_dequant_gemm_w4a8.py
- examples/dequantize_gemm/example_dequant_gemv_fp16xint4.py
- examples/dequantize_gemm/example_dequant_groupedgemm_bf16_mxfp4_hopper.py
- examples/dynamic_shape/example_dynamic.py
- examples/dynamic_shape/regression_example_dynamic.py
- examples/elementwise/example_elementwise_add.py
- examples/flash_attention/example_gqa_bwd.py
- examples/flash_attention/example_gqa_bwd_tma_reduce_varlen.py
- examples/flash_attention/example_mha_bwd_bhsd.py
- examples/flash_attention/example_mha_bwd_bshd.py
- examples/flash_attention/example_mha_bwd_bshd_wgmma_pipelined.py
- examples/flash_attention/example_mha_fwd_bhsd.py
- examples/flash_attention/example_mha_fwd_bshd_wgmma_pipelined.py
- examples/flash_attention/example_mha_fwd_varlen.py
- examples/flash_decoding/example_gqa_decode.py
- examples/flash_decoding/example_mha_inference.py
- examples/fusedmoe/example_fusedmoe_tilelang.py
- examples/gemm/example_gemm_intrinsics.py
- examples/gemm/example_gemm_persistent.py
- examples/gemm/example_gemm_schedule.py
- examples/gemm_fp8/example_tilelang_gemm_fp8.py
- examples/gemm_fp8/example_tilelang_gemm_fp8_2xAcc.py
- examples/gemm_fp8/example_tilelang_gemm_fp8_intrinsic.py
- examples/gemm_splitk/example_tilelang_gemm_splitk.py
- examples/gemm_splitk/example_tilelang_gemm_splitk_vectorize_atomicadd.py
- examples/gemm_streamk/example_tilelang_gemm_streamk.py
- examples/gemv/example_gemv.py
- examples/warp_specialize/example_warp_specialize_gemm_copy_0_gemm_1.py
artifact_dir: evidence/pull-bundles/tilelang/gh-1568
head_ref_oid: 31a02404f4cf5e6952004d4bc6ec058ef6035668
---
# [Refactor] Simplify do_bench calls by using default warmup and rep parameters

- Repository: `tile-ai/tilelang`
- PR: [1568](https://github.com/tile-ai/tilelang/pull/1568)
- Decision: `Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.`
- Reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.

## Changed Paths

- `examples/attention_sink/example_gqa_sink_bwd_bhsd.py`
- `examples/attention_sink/example_gqa_sink_fwd_bhsd_wgmma_pipelined.py`
- `examples/attention_sink/example_mha_sink_bwd_bhsd.py`
- `examples/attention_sink/example_mha_sink_fwd_bhsd.py`
- `examples/attention_sink/example_mha_sink_fwd_bhsd_wgmma_pipelined.py`
- `examples/blocksparse_attention/example_tilelang_block_sparse_attn.py`
- `examples/blocksparse_attention/example_tilelang_sparse_gqa_decode_paged.py`
- `examples/blocksparse_attention/example_tilelang_sparse_gqa_decode_varlen_indice.py`
- `examples/blocksparse_attention/example_tilelang_sparse_gqa_decode_varlen_mask.py`
- `examples/blocksparse_gemm/example_blocksparse_gemm.py`
- `examples/cast/example_per_token_cast_to_fp8.py`
- `examples/deepseek_mla/example_mla_decode.py`
- `examples/deepseek_nsa/example_tilelang_nsa_decode.py`
- `examples/deepseek_nsa/example_tilelang_nsa_fwd.py`
- `examples/deepseek_v32/fp8_lighting_indexer.py`
- `examples/deepseek_v32/sparse_mla_bwd.py`
- `examples/deepseek_v32/sparse_mla_fwd.py`
- `examples/deepseek_v32/sparse_mla_fwd_pipelined.py`
- `examples/deepseek_v32/topk_selector.py`
- `examples/dequantize_gemm/example_dequant_gemm_bf16_fp4_hopper.py`
- `examples/dequantize_gemm/example_dequant_gemm_bf16_mxfp4_hopper.py`
- `examples/dequantize_gemm/example_dequant_gemm_fp4_hopper.py`
- `examples/dequantize_gemm/example_dequant_gemm_w4a8.py`
- `examples/dequantize_gemm/example_dequant_gemv_fp16xint4.py`
- `examples/dequantize_gemm/example_dequant_groupedgemm_bf16_mxfp4_hopper.py`
- `examples/dynamic_shape/example_dynamic.py`
- `examples/dynamic_shape/regression_example_dynamic.py`
- `examples/elementwise/example_elementwise_add.py`
- `examples/flash_attention/example_gqa_bwd.py`
- `examples/flash_attention/example_gqa_bwd_tma_reduce_varlen.py`
- `examples/flash_attention/example_mha_bwd_bhsd.py`
- `examples/flash_attention/example_mha_bwd_bshd.py`
- `examples/flash_attention/example_mha_bwd_bshd_wgmma_pipelined.py`
- `examples/flash_attention/example_mha_fwd_bhsd.py`
- `examples/flash_attention/example_mha_fwd_bshd_wgmma_pipelined.py`
- `examples/flash_attention/example_mha_fwd_varlen.py`
- `examples/flash_decoding/example_gqa_decode.py`
- `examples/flash_decoding/example_mha_inference.py`
- `examples/fusedmoe/example_fusedmoe_tilelang.py`
- `examples/gemm/example_gemm_intrinsics.py`
- `examples/gemm/example_gemm_persistent.py`
- `examples/gemm/example_gemm_schedule.py`
- `examples/gemm_fp8/example_tilelang_gemm_fp8.py`
- `examples/gemm_fp8/example_tilelang_gemm_fp8_2xAcc.py`
- `examples/gemm_fp8/example_tilelang_gemm_fp8_intrinsic.py`
- `examples/gemm_splitk/example_tilelang_gemm_splitk.py`
- `examples/gemm_splitk/example_tilelang_gemm_splitk_vectorize_atomicadd.py`
- `examples/gemm_streamk/example_tilelang_gemm_streamk.py`
- `examples/gemv/example_gemv.py`
- `examples/warp_specialize/example_warp_specialize_gemm_copy_0_gemm_1.py`

## Evidence Bundle

`evidence/pull-bundles/tilelang/gh-1568`
