| Update moe_compute_expert_tokens、moe_finalize_routing_v2、moe_finalize_routing_v2_grad、moe_fused_topk、moe_gating_top_k、moe_gating_top_k_softmax、moe_gating_top_k_softmax_v2
Co-authored-by: JimmyLam2<linjingmin2@huawei.com>
# message auto-generated for no-merge-commit merge:
!1044 merge master2 into master
Update moe_compute_expert_tokens、moe_finalize_routing_v2、moe_finalize_routing_v2_grad、moe_fused_topk、moe_gating_top_k、moe_gating_top_k_softmax、moe_gating_top_k_softmax_v2
Created-by: JimmyLam2
Commit-by: JimmyLam2
Merged-by: cann-robot
Description: ## 描述
## 描述
Update moe_compute_expert_tokens、moe_finalize_routing_v2、moe_finalize_routing_v2_grad、moe_fused_topk、moe_gating_top_k、moe_gating_top_k_softmax、moe_gating_top_k_softmax_v2
## 关联的Issue
<!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000-->
<!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。-->
## 测试
<!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。-->
## 文档更新
<!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。-->
## 类型标签
<!-- [x] 表示选中 -->
- [ ] Bug修复
- [ ] 新特性
- [ ] 性能优化
- [ ] 文档更新
- [ ] 其他,请描述:
See merge request: cann/ops-transformer!1044 | 4 个月前 |