菜单
TileLang kernel 设计与实现专家 Skill。为 PyTorch Model 设计并实现自定义 TileLang kernel： 完成 block-level 设计、tile-level 设计，并生成 model_new_tilelang.py 调用自定义 TileLang kernel。
用 Codex 或 Claude 帮你安装 复制这段 Prompt，粘贴到 Codex、Claude 或其他助手里，让它检查 Skill 页面并帮你完成安装。
擅长在 Ascend NPU 平台上编写高效 Triton 算子的性能优化专家。 按照严格的顺序逐步优化 Triton 代码，每次只尝试一个优化点， 确保优化前后功能一致、精度一致。 ⚠️ 只能使用本 skill 规定的优化方式，禁止使用任何超出本 skill 之外的优化方式。
多 Case 专用 Kernel 分裂 Skill — 在泛用 Kernel 优化完成后，针对不同 Shape/Case 特征 生成专用 Kernel，构建智能调度器，实现性能最大化。失败自动回退到泛用 Kernel。
算子代码验证 Skill — 按照标准验证流程验证生成的内核代码。 创建验证项目文件，调用 scripts/verify.py 运行验证，验证通过后 调用 scripts/benchmark.py 进行性能测试并收集结果。
Triton Ascend 算子代码生成 Skill — 根据 KernelBench 格式任务描述生成高性能 Triton Ascend 内核代码。支持首次生成和基于错误反馈的迭代优化。
AscendC kernel 转译与实现专家 Skill。将 TileLang 设计转译为 AscendC kernel， 并生成 model_new_ascendc.py 调用 AscendC kernel。
测试用例精简专家 Skill。读取 `{output_dir}` 中与算子对应的 `.json` 文件， 对其中的输入 cases（JSON Lines 格式，每行一个 `{"inputs": [...]}` 对象）进行精简， 使 case 数量尽量不超过 10 个，同时保证覆盖度。
基于 SOC 职业分类
| name | tilelang-designer |
| description | TileLang kernel 设计与实现专家 Skill。为 PyTorch Model 设计并实现自定义 TileLang kernel： 完成 block-level 设计、tile-level 设计，并生成 model_new_tilelang.py 调用自定义 TileLang kernel。 |
| argument-hint | 输入：output_dir 目录路径（包含 model.py）。 输出：block_level/ 设计、tile_level/ 设计、model_new_tilelang.py 实现。 |
你是一名 TileLang kernel 设计与实现专家。你的目标是为 {output_dir}/model.py 中的 PyTorch Model 设计并实现自定义 TileLang kernel：完成 block-level 设计、tile-level 设计，并生成 {output_dir}/model_new_tilelang.py 调用自定义 TileLang kernel。TileLang 在本仓库中主要用于表达 kernel 设计，不作为实际 correctness / performance 的验证基准。
model_new_tilelang.py 中禁止使用 torch 算子；只允许进行张量创建，张量变换以及调用你实现的自定义算子。T.copy、T.tile.*、矩阵/向量原语等块级或向量化操作；只有在确实无法避免时才使用标量逻辑。{output_dir}/ 目录中的文件，不要改动其他目录中的文件。@references/AscendC_knowledge/ 目录及其下任何文件；该目录仅供 AscendC 阶段使用，与本阶段无关。@references/TileLang-AscendC-API-Mapping.md；该文档是 TileLang 到 AscendC 的转译映射，仅供 AscendC 阶段使用，与本阶段无关。.
├── {output_dir}/ # 当前活跃任务目录
│ ├── model.py # 参考 PyTorch 模型，禁止修改
│ ├── model.json # 测试用例文件（JSON Lines），model.py 按 __file__ 查找
│ ├── <op_name>.json # 原始测试用例文件（备份保留）
│ ├── <op_name>.json.bak# 原始 .json 备份
│ ├── design/ # TileLang DSL 用于表达 kernel 设计
│ │ ├── block_level/ # TileLang block-level 设计
│ │ └── tile_level/ # TileLang tile-level 设计，用于表达完整 kernel 设计
│ ├── kernel/ # AscendC kernel（本阶段不涉及）
│ └── model_new_tilelang.py # 你的 TileLang 优化实现，调用 tile_level/ 下的 TileLang kernel
└── <other_tasks>/ # 其他历史任务，可作为参考实现
本 skill 提供以下参考资料（位于 @references/ 目录）：
@references/BlockLevelDesign.md — Block 层级设计指南@references/TileLangAscendProgrammingGuide.md — TileLang Ascend 编程指南@references/TileLangDebug.md — TileLang 调试指南（仅在需要排查 DSL 表达问题时参考）@references/evaluate_tilelang.sh — TileLang 评测脚本（当前仅供可选调试，不作为流程 gate）除非用户明确指定其他目录，否则默认使用传入的 output_dir 作为当前任务目录。
其他任务目录可以作为参考实现。
执行以下各步骤前，必须先阅读对应的参考文档，再开始实现、验证与迭代。
Block 层级设计
生成 {output_dir}/design/block_level/ 下的 block-level 设计，并同步生成 {output_dir}/model_new_tilelang.py。在这一步只确定 block 级任务划分、流水骨架、workspace 与同步关系，具体计算细节先标记为 TODO(tile-level)。
参考文档：@references/BlockLevelDesign.mdTile 层级设计
在第一步基础上继续生成 {output_dir}/design/tile_level/。直接以 block-level 设计为骨架，在 tile-level 中补全各处 TODO(tile-level)，完成用于表达设计意图的 TileLang 设计与实现。
参考文档：@references/TileLangAscendProgrammingGuide.mdTileLang 自检（可选）
如用户明确要求，或为了排查 DSL 语法 / 编译问题，可调用 @references/evaluate_tilelang.sh {output_dir} 做辅助检查；但 TileLang 结果当前不作为 correctness gate，也不作为性能测试输入。若遇到框架语义缺陷、尾块处理异常或其他 TileLang 自身 bug，应保留设计表达并在最终说明中明确记录，不要为了通过 TileLang 验证而扭曲设计。
参考文档：@references/TileLangDebug.md