-
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
[XPU] bind Addmm #68560
[XPU] bind Addmm #68560
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
} | ||
} // namespace phi | ||
|
||
#ifdef PADDLE_WITH_XPU_XRE5 |
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_op_list控制不同的数据类型
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.
试过了,失败之后才换成现在这样
} // namespace phi | ||
|
||
#ifdef PADDLE_WITH_XPU_XRE5 | ||
PD_REGISTER_KERNEL(addmm, |
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.
同理
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 Category
Custom Device
PR Types
Others
Description
Bind Addmm.