mkldnn emitter - fixed memory leak. (#608)

* fixed memory leak. * clang format.

mkldnn emitter - fixed memory leak. (#608)
* fixed memory leak. * clang format.
0acaea58 · Louis Feng · Scott Cyphers · 95312b8e · 0acaea58 · 0acaea58
Commit 0acaea58 authored Mar 09, 2018 by Louis Feng Committed by Scott Cyphers Mar 09, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 0 deletions

mkldnn_emitter.cpp src/ngraph/runtime/cpu/mkldnn_emitter.cpp +6 -0

mkldnn_emitter.hpp src/ngraph/runtime/cpu/mkldnn_emitter.hpp +2 -0

No files found.
--- a/src/ngraph/runtime/cpu/mkldnn_emitter.cpp
+++ b/src/ngraph/runtime/cpu/mkldnn_emitter.cpp
@@ -25,6 +25,12 @@

 using namespace ngraph::runtime::cpu;

+MKLDNNEmitter::~MKLDNNEmitter()
+{
+    for (auto p : m_mkldnn_primitives)
+        delete p;
+}
+
 const std::vector<mkldnn::primitive*>& MKLDNNEmitter::get_mkldnn_primitives() const
 {
    return m_mkldnn_primitives;

--- a/src/ngraph/runtime/cpu/mkldnn_emitter.hpp
+++ b/src/ngraph/runtime/cpu/mkldnn_emitter.hpp
@@ -39,6 +39,8 @@ namespace ngraph
            {
            public:
                MKLDNNEmitter() {}
+                ~MKLDNNEmitter();
+
                const std::vector<mkldnn::primitive*>& get_mkldnn_primitives() const;

                size_t insert_primitive(mkldnn::primitive* primitive);