[oneDNN] Concat refactoring and disabling caching #35002
[oneDNN] Concat refactoring and disabling caching #35002jczaja merged 18 commits intoPaddlePaddle:developfrom
Conversation
|
Thanks for your contribution! |
jakpiase
left a comment
There was a problem hiding this comment.
I really like the initiative to change from factory to acquireAPI :)
There was a problem hiding this comment.
| ConcatMKLDNNHandler(const paddle::framework::ExecutionContext& ctx, | |
| ConcatMKLDNNHandler(const framework::ExecutionContext& ctx, |
There was a problem hiding this comment.
| paddle::framework::ToMKLDNNDataType(inputs[0]->type()); | |
| framework::ToMKLDNNDataType(inputs[0]->type()); |
There was a problem hiding this comment.
| for (size_t i = 0; i < inputs.size(); i++) { | |
| for (size_t i = 0; i < inputs.size(); ++i) { |
There was a problem hiding this comment.
| paddle::framework::vectorize<int64_t>(inputs[i]->dims()); | |
| framework::vectorize<int64_t>(inputs[i]->dims()); |
There was a problem hiding this comment.
| auto dst_dims = paddle::framework::vectorize<int64_t>(output->dims()); | |
| auto dst_dims = framework::vectorize<int64_t>(output->dims()); |
There was a problem hiding this comment.
| const std::vector<mkldnn::memory::desc>& srcs_md) { | |
| const std::vector<memory::desc>& srcs_md) { |
There was a problem hiding this comment.
| const mkldnn::memory::desc& dst_md, const int concat_axis, | |
| const memory::desc& dst_md, const int concat_axis, |
There was a problem hiding this comment.
| const framework::Tensor& input, int i) { | |
| const Tensor& input, int i) { |
There was a problem hiding this comment.
| ctx.template device_context<paddle::platform::MKLDNNDeviceContext>(); | |
| ctx.template device_context<platform::MKLDNNDeviceContext>(); |
There was a problem hiding this comment.
You have forgotten about this one
|
@jakpiase Please continue your review |
jakpiase
left a comment
There was a problem hiding this comment.
LGTM, I really appreciate your effort to unify our operators
bc89f59 to
4ad092a
Compare
PR types
Others
PR changes
OPs
Describe
This PR disabled oneDNN primitives caching by PaddlePaddle in favour of oneDNN its own caching mechanism. Also refactoring to Acquire API was made for concat