Skip to content
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

Merged
merged 6 commits into from
Feb 27, 2023

Conversation

houj04
Copy link
Contributor

@houj04 houj04 commented Feb 22, 2023

PR types

New features

PR changes

OPs

Describe

  • shape算子添加fp16的注册。细节修改了下单测。
  • assign算子添加double的注册。
  • lookup_table_v2算子添加fp16的注册。重写了单测。
  • ignore some failed test for KL2 #41342 这个PR中,修改了tools/check_file_diff_approvals.sh里面针对@skip_check_grad_ci的可以执行approve操作的人的列表,但是忘记加id了。本PR补上了。

@houj04 houj04 changed the title [XPU] add fp16 support for shape op. [XPU] add fp16 support for shape and lookup_table_v2 op. Feb 22, 2023
@houj04 houj04 closed this Feb 22, 2023
@houj04 houj04 reopened this Feb 22, 2023

class XPUTestLookupTableOP(XPUOpTestWrapper):
Copy link
Contributor

Choose a reason for hiding this comment

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

好像少了几个单测?TestLookupTableWIsSelectedRows和TestEmbedOpError都没有了

Copy link
Contributor Author

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以及子类。

Copy link
Contributor

@XiaociZhang XiaociZhang left a comment

Choose a reason for hiding this comment

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

LGTM

ZibinGuo
ZibinGuo previously approved these changes Feb 23, 2023
Copy link
Contributor

@ZibinGuo ZibinGuo left a comment

Choose a reason for hiding this comment

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

LGTM

QingshuChen
QingshuChen previously approved these changes Feb 23, 2023
@houj04 houj04 dismissed stale reviews from QingshuChen and ZibinGuo via ec539d5 February 27, 2023 03:06
@houj04 houj04 merged commit d2a0577 into PaddlePaddle:develop Feb 27, 2023
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.

4 participants