| feat: AlltoAllMatmul AICPU适配 新增通信引擎参数comm_mode
Co-authored-by: tangjn<tangjianing@huawei.com>
# message auto-generated for no-merge-commit merge:
!3118 merge master into master
feat: AlltoAllMatmul AICPU适配 新增通信引擎参数comm_mode
Created-by: tangjn
Commit-by: tangjn
Merged-by: ascend-robot
Description: feat: 新增通信引擎参数
为了在AlltoAllMatmul和AlltoAllQuantMatmul算子中,支持用户指定HCCL通信引擎(AICPU或CCU),为算子PTA接口新增一个可选入参comm_mode。
[#2777](https://gitcode.com/cann/ops-transformer/issues/2777)
See merge request: Ascend/torchair!3118 | 4 天前 |