MKLDNN 1.0.2 #2102
MKLDNN 1.0.2 #2102
Conversation
|
/azp run |
|
Azure Pipelines successfully started running 20 pipeline(s). |
|
/azp run |
|
Azure Pipelines successfully started running 20 pipeline(s). |
| @@ -197,7 +170,7 @@ std::vector<std::unique_ptr<ComputeCapability>> MKLDNNExecutionProvider::GetCapa | |||
| ORT_UNUSED_PARAMETER(kernel_registries); | |||
|
|
|||
| // temporary switch to toggle between mkldnn-vanilla and mkldnn-subgraph implementation using | |||
| // ORT_MKLDNN_SUBGRAPH environment variable | |||
| // | |||
| @@ -27,7 +27,7 @@ class MklDnnRelu : public MklDnnKernel { | |||
| OrtKernelContext* context, | |||
| mkldnn::engine& cpu_engine, | |||
| std::vector<mkldnn::primitive>& net, | |||
| mkldnn::memory::format& source_format) { | |||
| std::vector<std::unordered_map<int, mkldnn::memory>> &net_args) { | |||
|
|
||
| void Reset() { | ||
| SubgraphVariables() { |
There was a problem hiding this comment.
why is this needed? i thought we got rid of it in #1740
|
looks like the windows build fails. |
Looking at it. cloned MKL-DNN 1.0.2 is failing. It builds for me on my dev box. |
|
|
||
| namespace onnxruntime { | ||
| namespace mkl_dnn { | ||
|
|
||
| #define MKLDNN_EP_LRU_CACHE_DEFAULT_SIZE 500 |
There was a problem hiding this comment.
this will be implemented later right? remove for now.
| x_shape.GetDims().begin(), x_shape.GetDims().end()); | ||
|
|
||
| ort_source_desc_ = mkldnn::memory::desc( | ||
| {src_dims}, MklDnnType<T>(), ort_source_format_); | ||
| source_desc_ = ort_source_desc_; | ||
| src_md_.reset(new mkldnn::memory::desc( |
There was a problem hiding this comment.
we can take this opportunity to clean up the smart pointer usage in all these files.
src_md_ = make_shared(...) is preferred and more efficient than reset(new T)
let's clean all this up. thanks.
There was a problem hiding this comment.
Most of the pointers are unique_ptr and we need make_unique.
I was about to push my changes. I will make these changes and then create a new PR
There was a problem hiding this comment.
Most of the pointers are unique_ptr and we need make_unique.
I was about to push my changes. I will make these changes and then create a new PR
let's do the cleanup as a separate PR then. priority is to get this one passing the CI's and merged soon.
There was a problem hiding this comment.
okay. I will close this and create new new PR. As I have merged latest from master repo.
|
Creating new PR |
Upgrading MKL-DNN EP to MKL-DNN 1.0.2