Wrap debug dump in LLVM_DEBUG

PiperOrigin-RevId: 267774506

1 files changed, 7 insertions, 4 deletions

diff --git a/mlir/lib/ExecutionEngine/ExecutionEngine.cpp b/mlir/lib/ExecutionEngine/ExecutionEngine.cpp
index f991d7dff17..2ba50544e51 100644
--- a/mlir/lib/ExecutionEngine/ExecutionEngine.cpp
+++ b/mlir/lib/ExecutionEngine/ExecutionEngine.cpp
@@ -36,10 +36,13 @@
 #include "llvm/ExecutionEngine/Orc/RTDyldObjectLinkingLayer.h"
 #include "llvm/ExecutionEngine/SectionMemoryManager.h"
 #include "llvm/IR/IRBuilder.h"
+#include "llvm/Support/Debug.h"
 #include "llvm/Support/Error.h"
 #include "llvm/Support/TargetRegistry.h"
 #include "llvm/Support/ToolOutputFile.h"
 
+#define DEBUG_TYPE "execution-engine"
+
 using namespace mlir;
 using llvm::dbgs;
 using llvm::Error;
@@ -77,12 +80,12 @@ void SimpleObjectCache::notifyObjectCompiled(const Module *M,
 std::unique_ptr<MemoryBuffer> SimpleObjectCache::getObject(const Module *M) {
   auto I = cachedObjects.find(M->getModuleIdentifier());
   if (I == cachedObjects.end()) {
-    dbgs() << "No object for " << M->getModuleIdentifier()
-           << " in cache. Compiling.\n";
+    LLVM_DEBUG(dbgs() << "No object for " << M->getModuleIdentifier()
+                      << " in cache. Compiling.\n");
     return nullptr;
   }
-  dbgs() << "Object for " << M->getModuleIdentifier()
-         << " loaded from cache.\n";
+  LLVM_DEBUG(dbgs() << "Object for " << M->getModuleIdentifier()
+                    << " loaded from cache.\n");
   return MemoryBuffer::getMemBuffer(I->second->getMemBufferRef());
 }