-
Notifications
You must be signed in to change notification settings - Fork 5.5k
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
[NPU] add NPU ops of compare, test=develop #34365
[NPU] add NPU ops of compare, test=develop #34365
Conversation
Thanks for your contribution! |
4d71fb0
to
c459554
Compare
… add_compare_npu_op_dev
… add_compare_npu_op_dev
def init_dtype(self): | ||
self.dtype = np.float16 | ||
|
||
|
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.
建议这里所有的单测case,参考test_compare_op.py用 create_test的方式重写一个,一个是增加代码简洁度,可以直接看到有哪几个OP的单测被cover了;另一个原因是NPU在设计单测上一个是要尽量和CPU/CUDA的单测对齐,并且单测case要>= CPU/CUDA的单测case来保证NPU算子的准确性。
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 types
New features
PR changes
OPs
Describe
Add new NPU compare ops of less_equal, greater_than, greater_equal.