Closed
[Bug-Report|缺陷反馈] trans_data算子补充rt2.0版本InferDataType功能 #4173
zhangquanxinAdd on 13 days agoClosed at 11 days ago
13 days ago Added label: bug-report
陈思
13 days ago Comments:
13 days ago Comments:
/assign @zhangquanxin


13 days ago Assign zhangquanxin as responsible person
11 days ago Close issue
11 days ago Added label: resolved
一、问题描述 (必填)
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