-
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
Logical Ops support more data types #34141
Conversation
✅ This PR's description meets the template requirements! |
Thanks for your contribution! |
cd2a3ba
to
af4943f
Compare
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
|
||
REGISTER_OP_NPU_KERNEL(logical_and, | ||
ops::LogicalAndPUKernel<plat::NPUDeviceContext, bool>); | ||
ops::LogicalAndPUKernel<plat::NPUDeviceContext, bool>, |
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.
LogicalAndPUKernel
请确认下这个类型名称;推理认为增加这些实例带来的体积增量 9 MB 有些大了,请给出一些原因吧,谢谢
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.
因宏展开,推理库体积增加9M,GPU 4arch 预计增加36M,总体积增加 4.3%
这个PR 比较独立,先approve通过,后续也会看情况需不需要处理
PR types
Bug fixes
PR changes
OPs
Describe
Solve issue #34133 .
Logical Ops now support
bool
,int8
,int16
,int32
,int64
,float
, anddouble
.