Tilelang Pass Design

TileLang-Ascend Pass 设计文档生成 Skill

1. 目标

根据 Pass 需求信息，生成一份完整的 TileLang-Ascend Pass 设计文档（pass-design.md），涵盖以下核心决策：

• Pass 定位：Phase 1 / Phase 2 归属、Pipeline 具体位置
• 依赖分析：上游 Pass 输入、下游 Pass 输出、数据流传递
• IR 变换设计：输入 IR 结构、输出 IR 结构、变换逻辑
• 实现方案：C++ 类设计、核心方法、Python Wrapper、Pipeline 集成
• 测试方案：功能测试、依赖测试、边界测试
• 风险分析：已知约束、常见错误、与其他 Pass 的交互

2. 输入要求

必需信息

提问规则（必须严格遵守）：

1. 每次只询问一个字段：使用 question 工具时，questions 数组中只包含一个元素
2. 按表格顺序依次询问：Pass 名称 → 功能描述 → IR 变换类型 → 平台范围
3. 已提供的字段跳过：如果用户在初始请求中已提供某个字段的值，跳过该字段继续下一个

推荐信息

3. 工作流程

Phase 1：需求澄清

1. 解析用户提供的 Pass 需求信息
2. 检查必需字段是否完整
3. 按顺序逐一提问补全缺失字段（每次只问一个）

Phase 2：信息收集

1. 查阅 Pass 定位参考资料：

   • tilelang-pass-workflow-analyzer/references/pass-pipeline-overview.md - Pipeline 架构
   • tilelang-pass-workflow-analyzer/references/new-pass-placement-guide.md - 定位决策流程
   • tilelang-pass-workflow-analyzer/references/pass-dependency-graph.md - 依赖关系

2. 查阅 Pass 实现参考资料：

   • tilelang-pass-analyzer/references/pass-registry-ascend.md - 类似 Pass 实现
   • tilelang-pass-analyzer/references/ir-examples.md - IR 变换示例格式

3. 查阅本 skill 的实现模式参考：

   • references/pass-impl-patterns.md - C++ 类模板、注册方式

Phase 3：Pass 定位分析

按照 new-pass-placement-guide.md 的决策流程：

Step 1：分析 Pass 功能

• Pass 的主要功能是什么？
• Pass 修改 IR 还是收集信息？
• Pass 是平台无关还是 Ascend 特定？
• Pass 属于哪类优化（Lowering / 合法化 / 内存 / 流水线 / 同步 / 其他）？

Step 2：分析依赖关系

• Pass 需要哪些输入数据（attrs）？
• 这些数据由哪个 Pass 产生？
• Pass 产生哪些输出数据？
• 这些数据由哪个 Pass 使用？

Step 3：确定阶段归属

• DSL Lowering / IR 合法化 → Phase 1
• 硬件优化 / 内存优化 / 同步优化 → Phase 2
• 输入数据来自 Phase 1 → 可在 Phase 1 或 Phase 2
• 输入数据来自 Phase 2 → 必须在 Phase 2

Step 4：确定具体位置

• 依赖优先：Pass 必须在产生其输入数据的 Pass 后执行
• 供给优先：Pass 必须在使用其输出数据的 Pass 前执行
• 功能相邻：将 Pass 放在功能相似的 Pass 附近

Phase 4：生成 pass-design.md

基于 templates/pass-design-template.md 模板，填充所有章节：

1. 概述
2. Pass 定位
3. IR 变换设计
4. 实现方案
5. 测试方案
6. 风险点与注意事项
7. 交付清单

Phase 5：质量自检

按照 §5 中的自检清单逐项检查，确保文档质量。

Phase 6：针对性修订

仅修正未通过自检的项目。信息确实不足的标注为「待确认」并说明原因。

Phase 7：输出与关联引导

1. 将 pass-design.md 输出到当前目录或用户指定路径
2. 若文件已存在，询问是否覆盖
3. 关联引导：提示用户可使用相关 skill 查看详细信息

4. 定位决策速查表

阶段归属速查

典型定位场景

关键数据依赖

5. 质量自检清单

生成 pass-design.md 后，逐项检查：

通过条件：所有必须项全部通过。

6. 信息源优先级

冲突处理：当信息源之间矛盾时，以 pass-pipeline-overview.md 和 new-pass-placement-guide.md 为准。

7. 错误处理

8. 完成报告与关联引导

文档生成完成后，输出以下格式的报告：

## Pass 设计文档生成报告

- Pass 名称: {Pass 名称}
- 阶段归属: {Phase 1 / Phase 2}
- Pipeline 位置: {具体位置描述}
- IR 变换类型: {修改 IR / 收集信息 / 验证 IR}
- 输出路径: {文件路径}

### 自检结果
1. 阶段归属明确: ✅ / ❌
2. Pipeline 位置具体: ✅ / ❌
3. 依赖关系完整: ✅ / ❌
4. IR 变换示例清晰: ✅ / ❌
5. C++ 实现方案明确: ✅ / ❌
6. Pipeline 集成代码完整: ✅ / ❌
7. 无占位符: ✅ / ❌

### 待确认项
- {列出需要用户进一步确认的内容}

### 后续步骤建议
1. 查看详细 Pass 工作流：使用 **tilelang-pass-workflow-analyzer** skill
2. 查看类似 Pass 实现：使用 **tilelang-pass-analyzer** skill
3. 开始实现 Pass：使用 **tilelang-pass-generate** skill

9. 参考资料

本 skill 依赖以下参考文件：

10. 与其他 Skill 的关系