-
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
move fusion_group kernel to phi #53781
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
❌ The PR is not created using PR's template. You can refer to this Demo. |
@From00 ,帮忙 Review 一下 ~ |
namespace phi { | ||
|
||
template <typename T, typename Context> | ||
void FusionGroupKernel(const Context& dev_ctx, |
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.
fused类算子需统一放置在paddle/phi/kernels/fused
目录下,代码实现在phi::fusion
命名空间中
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,但是我看 paddle/phi/kernels/fused
下的 kernel 都没有对应的头文件,所以我把头文件给删了
phi::FusionGroupKernel, | ||
float, | ||
double, | ||
phi::dtype::float16) {} |
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.
此算子需要迁移infershape,并在infershape中通过outs_dtype
属性推导输出类型。
可以在这个PR合入后单独提一个迁移infershape的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.
Get,后续单独提一个 PR
@From00 ,帮忙 Rerun 一下 PR-CI-CINN ~ |
@From00 ,帮忙再 Review 一下 ~ |
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
PR types
Others
PR changes
Others
Description