960dt芯片支持mma2a&a2amm算子ccu通信引擎
目前的信息还不足以确认 960dt 上 mma2a、a2amm 的 CCU 支持情况。现有片段展示的是面向 910B/910_93 的部分 MTE 通信实现,不能据此判断 960dt 的 CCU 适配状态,也不能将本需求直接认定为缺陷。
请补充 mma2a、a2amm 对应的完整算子或接口名称,以及目标 CANN 版本,以明确适配范围。若目标实现尚未接入 CCU,可考虑在通信引擎选择与任务执行处增加适配,重点确认通信同步和计算依赖关系,并评估与现有通信路径的兼容性。
PR #15055 针对 960dt 的 AlltoAllMatmul 和 MatmulAlltoAll 算子增加了 CCU 通信支持。相关改动包括:移除 AlltoAllMatmul 在 960dt 上固定使用 AICPU 通信引擎的配置,并在算法选择时传入芯片架构、增加算法名为空的异常检查。
关联 PR:PR #15055
Backgroud(背景信息)
960dt芯片支持mma2a&a2amm算子ccu通信引擎
Benefit / Necessity (价值/作用)
Design(设计方案)