New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[PIR] Add Dtype Transfer for pir #58397
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
|
||
OP_GET_KERNEL_TYPE_FOR_VAR_TEMPLATE = """ | ||
phi::DataType {op_name}::GetKernelTypeForVar( | ||
const std::string& var_name, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
var_name 有必要么?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
有必要
//pd_op.cc
// deal skip data transform
if (var_name == "out_size" || var_name == "size_tensor" || var_name == "scale_tensor"){
return expected_kernel_dtype;
}
paddle/phi/api/yaml/op_compat.yaml
Outdated
@@ -1866,6 +1866,9 @@ | |||
outputs : | |||
out : Y | |||
|
|||
- op : matmul | |||
complex_promote : [X, Y] |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
下面的 matmul 已经有该标记了
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
已删除
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
* add log * add getkerneltype func by yaml * delete VLOG * update * change kernelkey to datatype * update * move util functions into pd_op_lower_to_kernel_pass
* add log * add getkerneltype func by yaml * delete VLOG * update * change kernelkey to datatype * update * move util functions into pd_op_lower_to_kernel_pass
* add log * add getkerneltype func by yaml * delete VLOG * update * change kernelkey to datatype * update * move util functions into pd_op_lower_to_kernel_pass
* add log * add getkerneltype func by yaml * delete VLOG * update * change kernelkey to datatype * update * move util functions into pd_op_lower_to_kernel_pass
PR types
Others
PR changes
Others
Description
PCard-67164
为pir添加Dtype Transfer功能,修复elementwise_add测试中complex和real类型运算报错的问题