| megamoe host a4w4 adapt
Co-authored-by: weixin_49300047<lijianhua56@huawei.com>
# message auto-generated for no-merge-commit merge:
!5007 merge megamoe_a4w4 into master
megamoe host a4w4 adapt
Created-by: weixin_49300047
Commit-by: LiJianHua;weixin_49300047
Merged-by: cann-robot
Description: ## 描述
<!--在这里详细描述你的改动,包括改动的原因和所采取的方法。-->
1. 在 aclnn_common.h 中补齐 FP4 / FP8 dtype 桥接与 HCCL 公共封装,作为 A4W4 适配的基础设施。
2. 在 mega_moe_tiling.cpp 中将参数命名语义化(N → hiddenDim)并对齐 A4W4 路径的类型签名。
3. 在 torch 适配层 mega_moe.cpp 中修复 weight1 真实 dtype 透传,保证 tiling 走入正确的 A4W4 分支。
4. 在 moe_distribute_dispatch_v2_tiling.cpp 中补齐 scalesDim1 维度校验,将非法 shape 拦截在 host 侧。
## 关联的Issue
<!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000-->
<!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。-->
https://gitcode.com/cann/ops-transformer/issues/2453
## 测试
<!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。-->
## 文档更新
<!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。-->
## 类型标签
<!-- [x] 表示选中 -->
- [ ] 🐛 Bug 修复
- [x] ✨ 新特性
- [ ] ⚡ 性能优化
- [ ] ♻️ 重构
- [ ] 🧪 测试
- [ ] 📦 构建/CI
- [ ] 🔧 配置变更
- [ ] 📝 文档更新
- [ ] ⬆️ 依赖升级
- [ ] 🔒 安全修复
- [ ] 🧹 代码清理
- [ ] ❓ 其他,请描述:
See merge request: cann/ops-transformer!5007 | 18 天前 |