Skip to content

[NPU] add batch_norm_op_npu and test#34056

Merged
qili93 merged 3 commits intoPaddlePaddle:developfrom
ronny1996:npu_batch_norm
Aug 11, 2021
Merged

[NPU] add batch_norm_op_npu and test#34056
qili93 merged 3 commits intoPaddlePaddle:developfrom
ronny1996:npu_batch_norm

Conversation

@ronny1996
Copy link
Contributor

@ronny1996 ronny1996 commented Jul 9, 2021

PR types

Others

PR changes

OPs

Describe

add batch_norm_op_npu and test

image

@paddle-bot-old
Copy link

paddle-bot-old bot commented Jul 9, 2021

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

Copy link
Contributor

Choose a reason for hiding this comment

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

16-19行的头文件已经被batch_norm_op.h和npu_op_runner.h包含,可以不需要?

Copy link
Contributor

Choose a reason for hiding this comment

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

fill_constant_op.h也可以不需要?

Copy link
Contributor

Choose a reason for hiding this comment

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

最好也参考 test_batch_norm_op.py把里面所有的test case加上,不能支持的就注释一下说明原因,尽量保证单测的coverage。

Comment on lines 165 to 173
Copy link
Contributor

Choose a reason for hiding this comment

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

Why add pinned allocator here? Is it needed in batch_norm op?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

It is no longer needed and has been removed

@ronny1996 ronny1996 requested review from qili93 and zhiqiu July 16, 2021 03:14
qili93
qili93 previously approved these changes Jul 16, 2021
Copy link
Contributor

@qili93 qili93 left a comment

Choose a reason for hiding this comment

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

LGTM

@ronny1996 ronny1996 changed the title add batch_norm_op_npu and test [NPU] add batch_norm_op_npu and test Jul 28, 2021
Copy link
Contributor

@zhiqiu zhiqiu left a comment

Choose a reason for hiding this comment

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

LGTM for ShareDataWith

Copy link
Contributor

@qili93 qili93 left a comment

Choose a reason for hiding this comment

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

LGTM

@qili93 qili93 merged commit 9ed5db2 into PaddlePaddle:develop Aug 11, 2021
@ronny1996 ronny1996 deleted the npu_batch_norm branch September 9, 2021 07:52
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.

3 participants