[Mlir-commits] [mlir] a1b6dae - [mlir][LLVMIR] Add LLVMDialect check in `DIScopeForLLVMFuncOp` (#110427)

[llvmlistbot at llvm.org]

Author: Longsheng Mou
Date: 2024-10-04T19:51:09+08:00
New Revision: a1b6daea2c85e2a835c7e9f982493b2c1b979c6e

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

LOG: [mlir][LLVMIR] Add LLVMDialect check in `DIScopeForLLVMFuncOp` (#110427)

This PR adds an LLVMDialect check in `DIScopeForLLVMFuncOp` to prevent
crashes. Fixes #108390.

Added: 
    

Modified: 
    mlir/lib/Dialect/LLVMIR/Transforms/DIScopeForLLVMFuncOp.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/LLVMIR/Transforms/DIScopeForLLVMFuncOp.cpp b/mlir/lib/Dialect/LLVMIR/Transforms/DIScopeForLLVMFuncOp.cpp
index 2cfaffa7c8efce..b276e06f93c612 100644
--- a/mlir/lib/Dialect/LLVMIR/Transforms/DIScopeForLLVMFuncOp.cpp
+++ b/mlir/lib/Dialect/LLVMIR/Transforms/DIScopeForLLVMFuncOp.cpp
@@ -91,6 +91,10 @@ struct DIScopeForLLVMFuncOp
     Location loc = module.getLoc();
 
     MLIRContext *context = &getContext();
+    if (!context->getLoadedDialect<LLVM::LLVMDialect>()) {
+      emitError(loc, "LLVM dialect is not loaded.");
+      return signalPassFailure();
+    }
 
     // To find a DICompileUnitAttr attached to a parent (the module for
     // example), otherwise create a default one.