add toggle to compiler diagnostic output (#388)

src/ngraph/codegen/compiler.cpp

@@ -123,6 +123,7 @@ static std::string GetExecutablePath(const char* Argv0)
 StaticCompiler::StaticCompiler()
     : m_precompiled_header_valid(false)
     , m_debuginfo_enabled(false)
+    , m_enable_diag_output((std::getenv("NGRAPH_COMPILER_DIAG_ENABLE") != nullptr))
     , m_source_name("code.cpp")
 {
     initialize();
@@ -150,19 +151,27 @@ void StaticCompiler::initialize()
     args.push_back("-inline-threshold=1000000");
 
     // Prepare DiagnosticEngine
-    IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions();
-    DiagOpts->ErrorLimit = 20;
-    TextDiagnosticPrinter* textDiagPrinter = new clang::TextDiagnosticPrinter(errs(), &*DiagOpts);
-    IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
-    DiagnosticsEngine DiagEngine(DiagID, &*DiagOpts, textDiagPrinter);
+    IntrusiveRefCntPtr<DiagnosticOptions> diag_options = new DiagnosticOptions();
+    diag_options->ErrorLimit = 20;
+    IntrusiveRefCntPtr<DiagnosticIDs> diag_id(new DiagnosticIDs());
+    DiagnosticsEngine diag_engine(diag_id, &*diag_options);
 
     // Create and initialize CompilerInstance
     m_compiler = std::unique_ptr<CompilerInstance>(new CompilerInstance());
-    m_compiler->createDiagnostics();
+    DiagnosticConsumer* diag_consumer;
+    if (m_enable_diag_output)
+    {
+        diag_consumer = new TextDiagnosticPrinter(errs(), &*diag_options);
+    }
+    else
+    {
+        diag_consumer = new IgnoringDiagConsumer();
+    }
+    m_compiler->createDiagnostics(diag_consumer);
 
     // Initialize CompilerInvocation
     CompilerInvocation::CreateFromArgs(
-        m_compiler->getInvocation(), &args[0], &args[0] + args.size(), DiagEngine);
+        m_compiler->getInvocation(), &args[0], &args[0] + args.size(), diag_engine);
 
     configure_search_path();
 

src/ngraph/codegen/compiler.hpp

@@ -86,6 +86,7 @@ private:
     std::unique_ptr<clang::CompilerInstance> m_compiler;
     bool m_precompiled_header_valid;
     bool m_debuginfo_enabled;
+    bool m_enable_diag_output;
     std::string m_source_name;
     std::vector<std::string> m_extra_search_path_list;
     std::string m_pch_path;