Commit ca4bed7
[PHI] Migrate pool2d and pool2d_grad kernels (#47423)
* 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]>1 parent 1831919 commit ca4bed7
File tree
5 files changed
+434
-405
lines changed- paddle
- fluid/operators/mkldnn
- phi
- backends/onednn
- kernels/onednn
5 files changed
+434
-405
lines changed
0 commit comments