Commit 0acaea58 authored by Louis Feng's avatar Louis Feng Committed by Scott Cyphers

mkldnn emitter - fixed memory leak. (#608)

* fixed memory leak.

* clang format.
parent 95312b8e
...@@ -25,6 +25,12 @@ ...@@ -25,6 +25,12 @@
using namespace ngraph::runtime::cpu; 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 const std::vector<mkldnn::primitive*>& MKLDNNEmitter::get_mkldnn_primitives() const
{ {
return m_mkldnn_primitives; return m_mkldnn_primitives;
......
...@@ -39,6 +39,8 @@ namespace ngraph ...@@ -39,6 +39,8 @@ namespace ngraph
{ {
public: public:
MKLDNNEmitter() {} MKLDNNEmitter() {}
~MKLDNNEmitter();
const std::vector<mkldnn::primitive*>& get_mkldnn_primitives() const; const std::vector<mkldnn::primitive*>& get_mkldnn_primitives() const;
size_t insert_primitive(mkldnn::primitive* primitive); size_t insert_primitive(mkldnn::primitive* primitive);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment