| [PLZ]MhcPre支持训练场景
Co-authored-by: LiWeiJian16<liweijian16@huawei.com>
# message auto-generated for no-merge-commit merge:
!4136 merge mhc_pre_fix_0331 into master
[PLZ]MhcPre支持训练场景
Created-by: liweijian16
Commit-by: LiWeiJian16
Merged-by: cann-robot
Description: ## 描述
<!--在这里详细描述你的改动,包括改动的原因和所采取的方法。-->
新增MhcPre正向算子支持训练场景,具体而言,在outflag=1时,输出inv_rms, h_mix, h_pre.
同时不再限制BS范围,理论支持长序列; D范围放宽至1-16384.
## 关联的Issue
<!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000-->
<!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。-->
https://gitcode.com/cann/ops-transformer/issues/1891
## 测试
<!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。-->
使用新精度2.1标准进行测试,通过
- [x] 网络用例
- [x] L2级用例
- [x] STC用例
## 文档更新
<!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。-->
更新了mhc/mhc_pre/README.md、mhc/mhc_pre/docs/aclnnMhcPre.md文件中的规格说明
## 类型标签
<!-- [x] 表示选中 -->
- [x] 🐛 Bug 修复
- [x] ✨ 新特性
- [ ] ⚡ 性能优化
- [ ] ♻️ 重构
- [x] 🧪 测试
- [ ] 📦 构建/CI
- [ ] 🔧 配置变更
- [x] 📝 文档更新
- [ ] ⬆️ 依赖升级
- [ ] 🔒 安全修复
- [ ] 🧹 代码清理
- [ ] ❓ 其他,请描述:
See merge request: cann/ops-transformer!4136 | 1 个月前 |