Unverified Commit a432b1a7 authored by Robert Kimball's avatar Robert Kimball Committed by GitHub

address potential double free memory (#1734)

parent 0051f201
...@@ -59,6 +59,10 @@ namespace ngraph ...@@ -59,6 +59,10 @@ namespace ngraph
void cleanup_runtime_context(); void cleanup_runtime_context();
protected: protected:
CPU_CallFrame(const CPU_CallFrame&) = delete;
CPU_CallFrame(CPU_CallFrame&&) = delete;
CPU_CallFrame& operator=(const CPU_CallFrame&) = delete;
std::shared_ptr<CPU_ExternalFunction> m_external_function; std::shared_ptr<CPU_ExternalFunction> m_external_function;
EntryPoint m_compiled_function; EntryPoint m_compiled_function;
CPURuntimeContext* ctx; CPURuntimeContext* ctx;
......
...@@ -55,4 +55,9 @@ public: ...@@ -55,4 +55,9 @@ public:
void* m_allocated_buffer_pool = nullptr; void* m_allocated_buffer_pool = nullptr;
size_t m_buffer_size; size_t m_buffer_size;
bool m_custom_memory; bool m_custom_memory;
private:
GPUTensor(const GPUTensor&) = delete;
GPUTensor(GPUTensor&&) = delete;
GPUTensor& operator=(const GPUTensor&) = delete;
}; };
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