[mlir][Transforms][NFC] GreedyPatternRewriteDriver: Reformat debug logic

[springerm]

Summary

Do not duplicate code that is performing actual work, put debug code around it.

[vitalybuka]

Is this by this one D144552 https://lab.llvm.org/buildbot/#/builders/238/builds/3575 ?

[vitalybuka]

In D151207#4369212, @vitalybuka wrote:

Is this by this one D144552 https://lab.llvm.org/buildbot/#/builders/238/builds/3575 ?

I see it's fixed 5e10a8c436eb7edd3cb8a865218a3b271a458634

mlir/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp

if (succeeded(folder.tryToFold(op))) {
       changed = true;
       continue;
     }
 
     // Try to match one of the patterns. The rewriter is automatically
     // notified of any necessary changes, so there is nothing else to do
     // here.
 #ifndef NDEBUG
-      auto canApply = [&](const Pattern &pattern) {
-        LLVM_DEBUG({
-          logger.getOStream() << "\n";
-          logger.startLine() << "* Pattern " << pattern.getDebugName() << " : '"
-                             << op->getName() << " -> (";
-          llvm::interleaveComma(pattern.getGeneratedOps(), logger.getOStream());
-          logger.getOStream() << ")' {\n";
-          logger.indent();
-        });
-        return true;
-      };
-      auto onFailure = [&](const Pattern &pattern) {
-        LLVM_DEBUG(logResult("failure", "pattern failed to match"));
-      };
-      auto onSuccess = [&](const Pattern &pattern) {
-        LLVM_DEBUG(logResult("success", "pattern applied successfully"));
-        return success();
-      };
+#else
+    function_ref<bool(const Pattern &)> canApply = {};
+    function_ref<void(const Pattern &)> onFailure = {};
+    function_ref<LogicalResult(const Pattern &)> onSuccess = {};
+#endif
 
-      LogicalResult matchResult =
-          matcher.matchAndRewrite(op, *this, canApply, onFailure, onSuccess);
-      if (succeeded(matchResult))
-        LLVM_DEBUG(logResultWithLine("success", "pattern matched"));
-      else
-        LLVM_DEBUG(logResultWithLine("failure", "pattern failed to match"));
-#else
-      LogicalResult matchResult = matcher.matchAndRewrite(op, *this);
-#endif
 
-      if (succeeded(matchResult)) {
-        changed = true;
-        ++numRewrites;
-      }
+      LLVM_DEBUG(logResultWithLine("success", "pattern matched"));
+    } else {
+      LLVM_DEBUG(logResultWithLine("failure", "pattern failed to match"));
   }
 
   return changed;
 }
 
 void GreedyPatternRewriteDriver::addToWorklist(Operation *op) {
   // Gather potential ancestors while looking for a "scope" parent region.
   SmallVector<Operation *, 8> ancestors;