fix memory leak (#2100)

b25f1714 · Jayaram Bobba · Scott Cyphers · dbc39fa4 · b25f1714
Commit b25f1714 authored Nov 21, 2018 by Jayaram Bobba Committed by Scott Cyphers Nov 21, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 5 deletions

quantization.cpp src/ngraph/runtime/cpu/builder/quantization.cpp +4 -5

No files found.
--- a/src/ngraph/runtime/cpu/builder/quantization.cpp
+++ b/src/ngraph/runtime/cpu/builder/quantization.cpp
@@ -229,11 +229,10 @@ namespace ngraph
                                    {input_desc, executor::global_cpu_engine},
                                    {result_desc, executor::global_cpu_engine},
                                    attr);
-                                ctx->mkldnn_primitives[quantize_index]->reset(
-                                    (new mkldnn::reorder(reorder_desc,
-                                                         *ctx->mkldnn_primitives[deps[0]],
-                                                         *ctx->mkldnn_primitives[deps[1]]))
-                                        ->get());
+                                *ctx->mkldnn_primitives[quantize_index] =
+                                    mkldnn::reorder(reorder_desc,
+                                                    *ctx->mkldnn_primitives[deps[0]],
+                                                    *ctx->mkldnn_primitives[deps[1]]);
                            }
                            cpu::mkldnn_utils::set_memory_ptr(ctx, deps[0], arg0_tensor);
                            cpu::mkldnn_utils::set_memory_ptr(ctx, deps[1], out_tensor);