Error out when kernel function is not found while translating GPU calls.

PiperOrigin-RevId: 269327909
llvm · Sep 16, 2019 · 1da0290 · 1da0290
1 parent 6755dfd
commit 1da0290
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mlir/lib/Conversion/GPUToCUDA/ConvertLaunchFuncToCudaCalls.cpp b/mlir/lib/Conversion/GPUToCUDA/ConvertLaunchFuncToCudaCalls.cpp
@@ -310,6 +310,10 @@ void GpuLaunchFuncToCudaCallsPass::translateGpuLaunchCalls(
   // represents the cubin at runtime.
   // TODO(herhut): This should rather be a static global once supported.
   auto kernelFunction = getModule().lookupSymbol<FuncOp>(launchOp.kernel());
+  if (!kernelFunction) {
+    launchOp.emitError("missing kernel function ") << launchOp.kernel();
+    return signalPassFailure();
+  }
   auto cubinGetter =
       kernelFunction.getAttrOfType<SymbolRefAttr>(kCubinGetterAnnotation);
   if (!cubinGetter) {