[llvm] 9b06f75 - Reapply 9cffabc68ca380be937e192be909feff7b144822

[Mircea Trofin via llvm-commits]

Author: Mircea Trofin
Date: 2023-02-02T08:44:39-08:00
New Revision: 9b06f756b2438684e0a3357d3f6fd12b50a2092b

URL: https://github.com/llvm/llvm-project/commit/9b06f756b2438684e0a3357d3f6fd12b50a2092b
DIFF: https://github.com/llvm/llvm-project/commit/9b06f756b2438684e0a3357d3f6fd12b50a2092b.diff

LOG: Reapply 9cffabc68ca380be937e192be909feff7b144822

This reverts commit 735f117f4d0deb9644d65c8fe8a80add058e7a2b.

Added: 
    

Modified: 
    llvm/lib/Analysis/InteractiveModelRunner.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Analysis/InteractiveModelRunner.cpp b/llvm/lib/Analysis/InteractiveModelRunner.cpp
index 2e9a906964b1..99b009b6616f 100644
--- a/llvm/lib/Analysis/InteractiveModelRunner.cpp
+++ b/llvm/lib/Analysis/InteractiveModelRunner.cpp
@@ -18,18 +18,10 @@
 
 using namespace llvm;
 
-#define _IMR_CL_VALS(T, N) clEnumValN(TensorType::N, #T, #T),
-
-static cl::opt<TensorType> DebugReply(
-    "interactive-model-runner-echo-type", cl::init(TensorType::Invalid),
-    cl::Hidden,
+static cl::opt<bool> DebugReply(
+    "interactive-model-runner-echo-reply", cl::init(false), cl::Hidden,
     cl::desc("The InteractiveModelRunner will echo back to stderr "
-             "the data received "
-             "from the host as the specified type (for debugging purposes)."),
-    cl::values(SUPPORTED_TENSOR_TYPES(_IMR_CL_VALS)
-                   clEnumValN(TensorType::Invalid, "disable", "Don't echo")));
-
-#undef _IMR_CL_VALS
+             "the data received from the host (for debugging purposes)."));
 
 InteractiveModelRunner::InteractiveModelRunner(
     LLVMContext &Ctx, const std::vector<TensorSpec> &Inputs,
@@ -83,7 +75,8 @@ void *InteractiveModelRunner::evaluateUntyped() {
     }
     InsPoint += *ReadOrErr;
   }
-  if (DebugReply != TensorType::Invalid)
-    dbgs() << tensorValueToString(OutputBuffer.data(), OutputSpec);
+  if (DebugReply)
+    dbgs() << OutputSpec.name() << ": "
+           << tensorValueToString(OutputBuffer.data(), OutputSpec) << "\n";
   return OutputBuffer.data();
 }