-
Notifications
You must be signed in to change notification settings - Fork 5.9k
add fp16 unittests for kl2 #36583
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
add fp16 unittests for kl2 #36583
Conversation
|
Thanks for your contribution! |
paddle/fluid/pybind/pybind.cc
Outdated
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.
是否可以在接口名里加上xpu信息来表明这个接口是专门给xpu设备用的?
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.
这个是为了GPU一致,GPU也是有这个接口,一样的名字,我这里就沿用了。
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.
顺便加个int64类型吧,api里已经支持了
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.
这int64的kl1不支持,应该先在api里面增加了 kl1的支持后,再增加绑定这个接口。现在kl1和kl2只是做了fp16的区分。
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.
已经增加了注册,单测是跑不了的。缺少mean的int64的支持,cpu的单测都没有跑int64
|
LGTM |
qili93
left a comment
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
Describe
add fp16 unittests for kl2
*test=kunlun