-
Notifications
You must be signed in to change notification settings - Fork 5.7k
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
[PHI]Rename some PHI Kernel #49470
[PHI]Rename some PHI Kernel #49470
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
"reshape_with_xshape", {"X"}, {"shape"}, {"Out", "XShape"}); | ||
return KernelSignature("reshape", {"X"}, {"shape"}, {"Out", "XShape"}); | ||
} | ||
} else { |
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.
没有xshape的分支是不需要分发到reshape_infer
的kernel?
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.
谢谢,已修改
paddle/phi/ops/compat/einsum_sig.cc
Outdated
@@ -31,7 +29,7 @@ KernelSignature EinsumGradOpArgumentMapping(const ArgumentMappingContext& ctx) { | |||
} | |||
} // namespace phi | |||
|
|||
PD_REGISTER_BASE_KERNEL_NAME(einsum, einsum_raw); | |||
PD_REGISTER_BASE_KERNEL_NAME(einsum, einsum); |
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.
done
CustomDevice需要改吗?如果需要的话,建议也一起改了 |
谢谢提醒,这个在另一个PR,已补充到描述中 |
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
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
* fix xpu unittest issue: zero_dim_tensor * deal with leftout issue introduced by #49470
* rename kernel * delete sig * modify code according comment * fix ci bugs
* fix xpu unittest issue: zero_dim_tensor * deal with leftout issue introduced by PaddlePaddle#49470
PR types
Others
PR changes
Others
Describe
规范化以下Kernel有关命名:
flatten_with_xshape,reshape_with_xshape,einsum_raw,prod_raw,squeeze_with_xshape,unsqueeze_with_xshape
异构设备的相关修改:PaddlePaddle/PaddleCustomDevice#355