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

[Phi] move truncated_gaussian_random kernel #39971

Merged
merged 3 commits into from
Feb 28, 2022

Conversation

windstamp
Copy link
Contributor

@windstamp windstamp commented Feb 27, 2022

PR types

Function optimization

PR changes

OPs

Describe

[Phi] move truncated_gaussian_random kernel

还有两个有待后续完善的地方:

  1. paddle/fluid/framework/generator.h 中的部分依赖暂未迁移。
  2. truncated_gaussian_random_op.h 文件暂未删除,被多个其它文件依赖,且包含 NPU 和 XPU 环境。

@paddle-bot-old
Copy link

Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

Copy link
Contributor

@MingMingShangTian MingMingShangTian left a comment

Choose a reason for hiding this comment

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

LGTM


namespace phi {

KernelSignature TruncatedGaussianRandomOpArgumentMapping(
Copy link
Contributor

Choose a reason for hiding this comment

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

这个ArgumentMapping感觉可以不写,直接使用默认op_proto生成的应该也能work,可以在后续的PR去掉试试

Copy link
Contributor Author

Choose a reason for hiding this comment

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

开始没写这个,会报错。后续 PR 再尝试看看是否可以去掉。

@windstamp windstamp merged commit 23aa7a3 into PaddlePaddle:develop Feb 28, 2022
@windstamp windstamp deleted the npu_dev_phi_20220227_2 branch February 28, 2022 05:45
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