TileWithAxis(Ascend950 AICore 实现)在极端 shape/属性下存在三类缺陷:
此外属性契约与 TBE/AICPU 基线不一致:axis 缺省 0(应为 1)、tiles 为可选(应为必选);且算子声明支持动态 shape 但缺少 InferShapeRange 输出范围推导。
• 昇腾硬件型号:Ascend950(DAV_3510) • CANN版本: 9.0 系列 • 操作系统:Linux
关联 PR:https://gitcode.com/cann/ops-math/pull/6647
/assign @tan_xin
一、问题描述 (必填)
TileWithAxis(Ascend950 AICore 实现)在极端 shape/属性下存在三类缺陷:
此外属性契约与 TBE/AICPU 基线不一致:axis 缺省 0(应为 1)、tiles 为可选(应为必选);且算子声明支持动态 shape 但缺少 InferShapeRange 输出范围推导。
二、环境信息 (可选)
• 昇腾硬件型号:Ascend950(DAV_3510)
• CANN版本: 9.0 系列
• 操作系统:Linux
三、重现步骤 (可选)
四、预期结果 (可选)
关联 PR:https://gitcode.com/cann/ops-math/pull/6647