Skip to content

Commit

Permalink
[PHI] Migrate pool2d and pool2d_grad kernels (#47423)
Browse files Browse the repository at this point in the history
* add extra attr property set

* add type_info for all context

* add onednn context to all context

* fix context compile error

* simplify conv kernel args

* pass runtime attr into dev_ctx

* fix marco error

* clear conv_grad_kernel extra args

* merge conv_grad_grad into conv_grad

* clear conv2d_grad_grad extra attrs

* clear yaml and eager extra attr

* fix conv1d error

* change to thread local

* fix npu compile failed

* try to fix windows compile failed

* add conv2d onednn phi kernel

* fix ci bugs (#36)

* fix compile bugs (#38)

* fix extra input transform bug (#39)

* support dynamic created attr (#40)

* reset extra info gen code

* rm conv_grad_grad kernel

* reimpl pass attr adapting

* add int attr support

* remove vector inputnames creating

* fix map at error

* Update paddle/phi/kernels/onednn/conv_grad_kernel.cc

Co-authored-by: Sławomir Siwek <[email protected]>

* remove useless extra attrs

* replace mkldnn_engine by onednn_engine

* Migrate pool+grad to PHI

* Update paddle/fluid/operators/mkldnn/test_mkldnn_op_nhwc.cc

Co-authored-by: Sławomir Siwek <[email protected]>

* Update paddle/phi/kernels/onednn/pool_grad_kernel.cc

Co-authored-by: Sławomir Siwek <[email protected]>

* Update paddle/phi/kernels/onednn/pool_kernel.cc

Co-authored-by: Sławomir Siwek <[email protected]>

Co-authored-by: Chen Weihang <[email protected]>
Co-authored-by: YuanRisheng <[email protected]>
Co-authored-by: Chen Weihang <[email protected]>
Co-authored-by: Sławomir Siwek <[email protected]>
  • Loading branch information
5 people authored Nov 4, 2022
1 parent 1831919 commit ca4bed7
Show file tree
Hide file tree
Showing 5 changed files with 434 additions and 405 deletions.
Loading

0 comments on commit ca4bed7

Please sign in to comment.