-
Notifications
You must be signed in to change notification settings - Fork 5.7k
[Phi] mv kernel #39861
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] mv kernel #39861
Conversation
Hi, It's a test PR, it will not trigger CI. If you want to trigger CI, please remove |
1c1a3df
to
f168115
Compare
f168115
to
ea81b5c
Compare
ea81b5c
to
9ef3ab6
Compare
9ef3ab6
to
588929a
Compare
} // namespace phi | ||
|
||
PD_REGISTER_KERNEL(mv_grad, CPU, ALL_LAYOUT, phi::MvGradKernel, float, double) { | ||
} |
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.
这个pre-commit处理的换行吗,有点别扭,不如直接在kernel 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.
是的,pre-commit 某种情况下会这样自动更新一下,后来我又改回来了,但是又会导致 CI (应该是 PR-CI-Static-Check) 失败,必须这样才能通过。
你们是否可以和 PR-CI-Static-Check 确认一下,这种看着确实有点别扭。
|
||
namespace phi { | ||
|
||
KernelSignature MvOpArgumentMapping(const ArgumentMappingContext& 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.
这个op挺标准的,可以不用写这个函数,建议后续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.
好的,后续 PR 修复
|
||
namespace phi { | ||
|
||
KernelSignature MvOpArgumentMapping(const ArgumentMappingContext& 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.
这里前向的Mapping映射应该可以不写,使用默认op_proto映射即可,可以在后续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.
好的,后续 PR 修复
PR types
New features
PR changes
OPs
Describe
[Phi] mv kernel