Skip to content

MKLDNN nodes factory improvements#2252

Merged
dmitry-gorokhov merged 1 commit intoopenvinotoolkit:masterfrom
vladislav-volkov:dev-mkldnn-nodes-factory
Sep 30, 2020
Merged

MKLDNN nodes factory improvements#2252
dmitry-gorokhov merged 1 commit intoopenvinotoolkit:masterfrom
vladislav-volkov:dev-mkldnn-nodes-factory

Conversation

@vladislav-volkov
Copy link
Copy Markdown
Contributor

This fix contains optimisations for MKLDNN nodes factory. Algorithm complexity for graph nodes creation reduced from O(N*N) to O(N). All changes are taken from PR 1741.

@vladislav-volkov vladislav-volkov requested review from a team September 15, 2020 14:16
@openvino-pushbot openvino-pushbot added category: CPU OpenVINO CPU plugin category: IE Tests OpenVINO Test: plugins and common labels Sep 15, 2020
return getType() == TensorIterator;
}
REG_MKLDNN_PRIM_FOR(MKLDNNTensorIteratorNode, TensorIterator);
REG_MKLDNN_PRIM_FOR(MKLDNNTensorIteratorNode, MKLDNNPlugin::TensorIterator);
Copy link
Copy Markdown

Choose a reason for hiding this comment

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

Just wondering why do we need MKLDNNPlugin:: here

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

Without explicit namespace it clashes with TensorIterator from inference-engine/src/legacy_api/include/legacy/ie_layers.h.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

maybe it's better to remove global using namespace InferenceEngine?

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

removed

@dmitry-gorokhov
Copy link
Copy Markdown

@ilyachur could you please take a look as well?

@vladislav-volkov vladislav-volkov force-pushed the dev-mkldnn-nodes-factory branch 2 times, most recently from 82dab25 to 22e924c Compare September 24, 2020 09:00
@dmitry-gorokhov dmitry-gorokhov merged commit 8f1ee05 into openvinotoolkit:master Sep 30, 2020
asomsiko pushed a commit to asomsiko/openvino that referenced this pull request Oct 13, 2020
mryzhov pushed a commit to mryzhov/openvino that referenced this pull request Dec 16, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

category: CPU OpenVINO CPU plugin category: IE Tests OpenVINO Test: plugins and common

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants