Avoid assigning to an unchecked Error.

Fixes tensorflow/mlir#97

PiperOrigin-RevId: 264743395

diff --git a/mlir/lib/Support/JitRunner.cpp b/mlir/lib/Support/JitRunner.cpp
index f88d9b8..40f1292 100644 (file)
--- a/mlir/lib/Support/JitRunner.cpp
+++ b/mlir/lib/Support/JitRunner.cpp
@@ -333,16 +333,20 @@ int mlir::JitRunnerMain(
   auto transformer = mlir::makeLLVMPassesTransformer(
       passes, optLevel, /*targetMachine=*/tmOrError->get(), optPosition);
 
-  Error error = make_string_error("unsupported function type");
-  if (mainFuncType.getValue() == "f32")
-    error = compileAndExecuteSingleFloatReturnFunction(
-        m.get(), mainFuncName.getValue(), transformer);
-  else if (mainFuncType.getValue() == "memrefs")
-    error = compileAndExecuteFunctionWithMemRefs(
-        m.get(), mainFuncName.getValue(), transformer);
-  else if (mainFuncType.getValue() == "void")
-    error = compileAndExecuteVoidFunction(m.get(), mainFuncName.getValue(),
-                                          transformer);
+  // Get the function used to compile and execute the module.
+  using CompileAndExecuteFnT = Error (*)(
+      ModuleOp, StringRef, std::function<llvm::Error(llvm::Module *)>);
+  auto compileAndExecuteFn =
+      llvm::StringSwitch<CompileAndExecuteFnT>(mainFuncType.getValue())
+          .Case("f32", compileAndExecuteSingleFloatReturnFunction)
+          .Case("memrefs", compileAndExecuteFunctionWithMemRefs)
+          .Case("void", compileAndExecuteVoidFunction)
+          .Default(nullptr);
+
+  Error error =
+      compileAndExecuteFn
+          ? compileAndExecuteFn(m.get(), mainFuncName.getValue(), transformer)
+          : make_string_error("unsupported function type");
 
   int exitCode = EXIT_SUCCESS;
   llvm::handleAllErrors(std::move(error),