Add flag to enable memory sanitizer (#393)

* cleanup in-memory header files

* add switch to enable memory sanitizer (works like valgrind)

* removed header file cleanup as it was causing a segfault on program termination

src/ngraph/CMakeLists.txt

@@ -122,6 +122,10 @@ find_package(Graphviz QUIET)
 if (GRAPHVIZ_FOUND)
     set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DGRAPHVIZ_FOUND")
 endif()
+if(NGRAPH_ADDRESS_SANITIZER)
+    message(STATUS "Enable Address Sanitizer")
+    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g -fsanitize=address -fno-omit-frame-pointer")
+endif()
 
 include_directories(
     "${NGRAPH_INCLUDE_PATH}"

src/ngraph/codegen/compiler.cpp

@@ -225,6 +225,18 @@ void StaticCompiler::initialize()
 
 StaticCompiler::~StaticCompiler()
 {
+    // This is causing a segfault after program terminates
+    // will address later
+    // if (m_compiler)
+    // {
+    //     PreprocessorOptions& preprocessor_options =
+    //         m_compiler->getInvocation().getPreprocessorOpts();
+    //     for (auto& x : preprocessor_options.RemappedFileBuffers)
+    //     {
+    //         delete x.second;
+    //     }
+    //     m_compiler = nullptr;
+    // }
 }
 
 bool StaticCompiler::is_version_number(const string& path)

test/CMakeLists.txt

@@ -106,6 +106,9 @@ include_directories(".")
 
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DCURDIR=\\\"${CMAKE_CURRENT_SOURCE_DIR}\\\"")
+if(NGRAPH_ADDRESS_SANITIZER)
+    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g -fsanitize=address -fno-omit-frame-pointer")
+endif()
 
 add_executable(unit-test ${SRC})