-
Notifications
You must be signed in to change notification settings - Fork 5.6k
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
Implement some basic op kernel #2971
Conversation
…Paddle into implement_basic_OpKernel
可以加一些python的单测 |
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
|
||
namespace paddle { | ||
namespace operators { | ||
|
||
template <typename Place> | ||
template <typename Place, typename T> |
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.
Should we have paddle::kernel::Mul
instead of paddle::operators::MulKernel
?
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.
I am thinking on it, and haven't get a conclusion yet. And I will have a discussion with colleagues.
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和kernel关系比较紧密,经常需要结合着一起看,和tf还是有点区别。另外tf分开放感觉也不是非常方便。
@jacquesqiao 已经加上单测了
|
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 after add Python unit test
No description provided.