| mhc_sinkhorn_backward perf optimization
Co-authored-by: liuxiao222<liuxiao222@huawei.com>
# message auto-generated for no-merge-commit merge:
!5377 merge master into master
mhc_sinkhorn_backward perf optimization
Created-by: liuxiao222
Commit-by: liuxiao222
Merged-by: cann-robot
Description: ## 描述
1.公共计算逻辑外提,减少vec计算
1.优化broadcast的写法,消除循环中的同步阻塞,提高指令并行度
## 关联的Issue
关联issue:https://gitcode.com/cann/ops-transformer/issues/2467
## 测试
批跑白盒用例,网络用例
精度pss,性能达标
## 文档更新
不涉及
## 类型标签
<!-- [x] 表示选中 -->
- [ ] 🐛 Bug 修复
- [ ] ✨ 新特性
- [x] ⚡ 性能优化
- [ ] ♻️ 重构
- [x] 🧪 测试
- [ ] 📦 构建/CI
- [ ] 🔧 配置变更
- [ ] 📝 文档更新
- [ ] ⬆️ 依赖升级
- [ ] 🔒 安全修复
- [ ] 🧹 代码清理
- [ ] ❓ 其他,请描述:
See merge request: cann/ops-transformer!5377 | 17 天前 |