reseat MKL memory allocators to ngraph functions

src/ngraph/runtime/cpu/cpu_mkl_allocator.cpp

@@ -21,3 +21,10 @@ ngraph::runtime::cpu::CPUAllocator::CPUAllocator()
     , m_byte_size(0)
 {
 }
+
+ngraph::runtime::cpu::CPUAllocator::CPUAllocator(size_t size)
+{
+    m_byte_size = size;
+    mkl::i_malloc = MallocHook;
+    mkl::i_free = FreeHook;
+}

src/ngraph/runtime/cpu/cpu_mkl_allocator.hpp

@@ -57,9 +57,13 @@ public:
     ~CPUAllocator();
 
 private:
-    char* m_buffer;
+    void* m_buffer;
     size_t m_byte_size;
 
-    inline void* MallocHook(size_t size) { m_buffer = static_cast<char*>(ngraph_malloc(size)); }
-    inline void FreeHook() { ngraph_free(m_buffer); }
+    static inline void* MallocHook(size_t size)
+    {
+        void* ptr = static_cast<void*>(ngraph_malloc(size));
+        return ptr;
+    }
+    static inline void FreeHook(void* ptr) { ngraph_free(ptr); }
 };