Closed
[Bug-Report|缺陷反馈] trans_data算子补充rt2.0版本InferDataType功能 #4173
zhangquanxinAdd on  13 days agoClosed at  11 days ago
zhangquanxin
zhangquanxinCollaborators
13 days ago Create

一、问题描述 (必填)

trans_data 算子漏掉了 rt2.0 版本的 InferDataType 推导功能:目前仅通过 IMPL_OP_INFERSHAPE(TransData).InferShape 注册了形状推导,未注册 InferDataType,导致图编译阶段无法完成输出数据类型推导。

整改方案:参考原 rt1.0 IMPLEMT_COMMON_INFERFUNC(TransDataInferShape) 中输出数据类型与输入数据类型保持一致的逻辑,新增 conversion/trans_data/op_graph/trans_data_graph_infer.cpp,通过 IMPL_OP(TransData).InferDataType 注册实现。

二、环境信息 (可选)

• 昇腾硬件型号:
• CANN版本:
• 操作系统:
• 其他相关软件版本:

三、重现步骤 (可选)

构造 TransData 单算子或包含 TransData 的图,在 rt2.0 图编译流程中执行数据类型推导。

四、预期结果 (可选)

TransData 算子完成 InferDataType 注册,输出 dst 的数据类型与输入 src 保持一致,图编译正常通过。

💡 备注(选填)
关联 PR:https://gitcode.com/cann/ops-math/merge_requests/6773

likedislike
zhangquanxinzhangquanxinCollaborators
13 days ago Added label: bug-report
陈思
陈思Collaborators
13 days ago Comments:

/assign @zhangquanxin

likedislike
CANN-robotCANN-robotCollaborators
13 days ago Assign zhangquanxin as responsible person
CANN-robotCANN-robotCollaborators
11 days ago Close issue
CANN-robotCANN-robotCollaborators
11 days ago Added label: resolved