Skip to content

Conversation

@taixiurong
Copy link
Contributor

PR types

Others

PR changes

Others

Describe

add fp16 unittests for kl2
*test=kunlun

@paddle-bot-old
Copy link

Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

是否可以在接口名里加上xpu信息来表明这个接口是专门给xpu设备用的?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这个是为了GPU一致,GPU也是有这个接口,一样的名字,我这里就沿用了。

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

顺便加个int64类型吧,api里已经支持了

Copy link
Contributor Author

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的区分。

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

已经增加了注册,单测是跑不了的。缺少mean的int64的支持,cpu的单测都没有跑int64

@tangzhiyi11
Copy link
Contributor

LGTM

Copy link
Contributor

@qili93 qili93 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@qili93 qili93 merged commit 6838a18 into PaddlePaddle:develop Oct 27, 2021
ghost pushed a commit to piotrekobi/Paddle that referenced this pull request Nov 3, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants