| 新增mambav2 推理prefill阶段所需的关键算子
Co-authored-by: Jing_Huang66<huangjing66@huawei.com>
# message auto-generated for no-merge-commit merge:
!459 merge mamba2 into master
新增mambav2 推理prefill阶段所需的关键算子
Created-by: Jing_Huang66
Commit-by: Jing_Huang66
Merged-by: cann-robot
Description: ## 描述
新增mambav2 推理prefill阶段所需的关键算子, 包含6个融合算子:
- mamba2_causal_conv1d
- mamba2_rmsnormgated
- mamba2_chunk_cumsum
- mamba2_chunk_state
- mamba2_chunk_state_passing
- mamba2_chunk_scan
## 关联的Issue
关联Issue [#221](https://gitcode.com/cann/ops-transformer/issues/221)
## 测试
各算子完成精度和性能测试，参考各算子的test脚本
## 类型标签
<!-- [x] 表示选中 -->
- [ ] Bug修复
- [x ] 新特性
- [ ] 性能优化
- [ ] 文档更新
- [ ] 其他，请描述：
See merge request: cann/ops-transformer!459 | 2 个月前 |
