Add additional nullptr check for isDialectOp

`getCalledFunction()` can return `nullptr`.
GPUOpen-Drivers · Jun 11, 2024 · ec61062 · ec61062
1 parent 4e785a2
commit ec61062
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/lib/TableGen/GenDialect.cpp b/lib/TableGen/GenDialect.cpp
@@ -302,7 +302,11 @@ void llvm_dialects::genDialectDefs(raw_ostream& out, RecordKeeper& records) {
     }
 
     bool $Dialect::isDialectOp(::llvm::CallInst& op) {
-      return isDialectOp(op.getCalledFunction()->getName());
+      ::llvm::Function *calledFunc = op.getCalledFunction(); 
+      if (!calledFunc)
+        return false;
+      
+      return isDialectOp(calledFunc->getName());
     }
 
     bool $Dialect::isDialectOp(::llvm::Function& func) {

diff --git a/test/example/generated/ExampleDialect.cpp.inc b/test/example/generated/ExampleDialect.cpp.inc
@@ -31,7 +31,11 @@ namespace xd {
     }
 
     bool ExampleDialect::isDialectOp(::llvm::CallInst& op) {
-      return isDialectOp(op.getCalledFunction()->getName());
+      ::llvm::Function *calledFunc = op.getCalledFunction(); 
+      if (!calledFunc)
+        return false;
+
+      return isDialectOp(calledFunc->getName());
     }
 
     bool ExampleDialect::isDialectOp(::llvm::Function& func) {