forked from ccf-ai-infra/TileOPs-Metax
72 lines
1.3 KiB
Markdown
72 lines
1.3 KiB
Markdown
<!--
|
||
语言：简体中文
|
||
英文模板：.github/PULL_REQUEST_TEMPLATE/operator-migration.en.md
|
||
|
||
PR 标题规范：
|
||
[算子名] optimize: 本次优化简短说明
|
||
[算子名] feat: 本次新增功能简短说明
|
||
|
||
请在 optimize/feat 中选择一个，不要同时保留。
|
||
-->
|
||
|
||
## 小组课题信息
|
||
|
||
课题名称：
|
||
|
||
课题完整简介：概括本组整体研究目标、目标算子、核心优化方向、预期产出成果。
|
||
|
||
### PR 简要描述
|
||
|
||
算子名称：
|
||
|
||
小组：XX 组
|
||
|
||
成员：XXX、XXX、XXX
|
||
|
||
改动类型：
|
||
|
||
- [ ] `feat`：新增算子或功能
|
||
- [ ] `optimize`：优化已有实现
|
||
|
||
### 1. 本次 PR 优化方案
|
||
|
||
简述采用的硬件优化手段，例如访存优化、流水线优化、分块策略或计算单元通信优化。
|
||
|
||
优化前的主要问题：
|
||
|
||
采用的优化方案：
|
||
|
||
关键代码或配置变更：
|
||
|
||
### 2. 精度验证
|
||
|
||
对比基准实现：
|
||
|
||
测试命令：
|
||
|
||
测试 shape/dtype：
|
||
|
||
误差范围（atol/rtol）：
|
||
|
||
验证结果：
|
||
|
||
### 3. 性能数据【必填】
|
||
|
||
测试环境（GPU、驱动、MACA、PyTorch、TileLang）：
|
||
|
||
性能复现命令：
|
||
|
||
优化前：
|
||
|
||
优化后：
|
||
|
||
加速比：
|
||
|
||
瓶颈分析（mcProfiler 观测结果）：
|
||
|
||
### 4. 提交自检清单
|
||
|
||
- [ ] 全部测试用例运行通过，精度达标
|
||
- [ ] 已删除临时调试打印、冗余测试代码
|
||
- [ ] 性能指标可本地复现，数据真实有效
|