-
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
[XPU] add fp16 support for shape and lookup_table_v2 op. #50773
Conversation
|
||
class XPUTestLookupTableOP(XPUOpTestWrapper): |
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.
好像少了几个单测?TestLookupTableWIsSelectedRows和TestEmbedOpError都没有了
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.
TestLookupTableWIsSelectedRows
这个类的基类是unittest.TestCase
,然后TestLookupTableWithTensorIdsWIsSelectedRows
的基类是TestLookupTableWIsSelectedRows
。这两个类,在python/paddle/fluid/tests/unittests/test_lookup_table_v2_op.py里面都有实现。
所以我特意删掉了,只按照基于XPUOpTest
的方式重写了TestLookupTableOPBase
以及子类。
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
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
shape
算子添加fp16的注册。细节修改了下单测。assign
算子添加double的注册。lookup_table_v2
算子添加fp16的注册。重写了单测。tools/check_file_diff_approvals.sh
里面针对@skip_check_grad_ci
的可以执行approve操作的人的列表,但是忘记加id了。本PR补上了。