tilelang-perf-optimization

name	tilelang-perf-optimization
description	TileLang 算子性能调优与潜在性能劣化模式检查。提供性能数据采集、瓶颈诊断、优化实施、效果验证能力；也用于生成或评审算子时对照常见性能劣化模式示例检查当前 kernel 代码。触发：算子精度通过后需要优化性能、性能不及预期时。

TileLang 性能优化

工作流程

Step 1: 基线采集（性能 + 精度）
  → Step 2: 算子类型判断
  → Step 3: 阅读参考文档并识别优化点（输出到 optimization_log.md）
  → Step 4: 逐项实施优化点
  → Step 5: 效果验证（性能 + 精度）

核心约束

• 逐项实施：每次 Edit 只改一个优化点，改完立即验证
• 精度优先：精度未通过禁止性能优化
• 性能验证：必须使用 msprof op，禁止用 Python/Torch 计时
• Host 轻量化：禁止 host 侧全量数据搬运（F.pad、.contiguous()、.to(dtype) 等），必须移入 kernel

参考文档

• 优化指南：optimization-guide.md
• 反模式清单：performance-antipatterns.md
• API 用法：tilelang-api-best-practices
• 编程模式：tilelang-programming-model-guide
• cube最佳实践：cube_optimization_path.md
• vector最佳实践：vector-practices/

---

执行步骤

Step 1: 基线采集

在 examples/{op_name}/ 下查找含 @tilelang.jit 的脚本，运行：

msprof op --kernel-name="main_kernel" --output=./msprof_output python ./examples/{op_name}/<script_name>.py

精度未通过 → 禁止后续步骤。