已关闭
[BUG] Ascend950 比较类与 max/min 算子在 host/device tensor 混合输入下数据类型推导与设备校验缺陷 #168
wangqi_ai创建于 6月15日关闭于 8月19日
6月15日 添加了label:bug
wangqi_ai
6月15日 评论:
6月15日 评论:
/assign @wangqi_ai


6月15日 将 wangqi_ai 设为负责人
7月7日 关联了看板:MindStudio ISSUE管理
8月19日 添加了label:resolved
一、问题描述
在 Ascend950 平台上,比较类算子(eq/ge/gt/le/lt/ne)及 maximum/minimum 在 host tensor 与 device tensor 混合输入(以及 0 维 CPU scalar tensor 包装)场景下,存在以下缺陷:
other.dim()==0 && !is_npu(other)走*Scalar分支)与IsCPUScalar转 scalar 逻辑,导致输出 dtype 推导与标杆(CPU/CUDA)不一致(典型如 ge/le 的类型推导、maximum/minimum 的计算 dtype 差异)。二、重现步骤
在 Ascend950 上构造混合输入并与 CPU/CUDA 标杆对比:
分别调用
torch.ge/le/gt/lt/eq/ne与torch.maximum/torch.minimum(含out=与 inplace 变体),对比输出 dtype / device / 数值。三、预期结果
关联 PR:https://gitcode.com/Ascend/op-plugin/pull/4715