[Mlir-commits] [mlir] 6da8d6c - [mlir][Linalg] NFC - Improve usage of mlir::linalg::isaContractionOpInterface

[Nicolas Vasilache]

Author: Nicolas Vasilache
Date: 2021-02-05T13:37:00Z
New Revision: 6da8d6c68f6f89c574fae480d79bb23b67e3bc85

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

LOG: [mlir][Linalg] NFC - Improve usage of mlir::linalg::isaContractionOpInterface

Added: 
    

Modified: 
    mlir/lib/Dialect/Linalg/IR/LinalgInterfaces.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Linalg/IR/LinalgInterfaces.cpp b/mlir/lib/Dialect/Linalg/IR/LinalgInterfaces.cpp
index 0aafc4fc1050..d256942e8d02 100644
--- a/mlir/lib/Dialect/Linalg/IR/LinalgInterfaces.cpp
+++ b/mlir/lib/Dialect/Linalg/IR/LinalgInterfaces.cpp
@@ -115,6 +115,8 @@ static MatchContractionResult isContractionInterfaceImpl(Operation *op) {
 }
 
 bool mlir::linalg::isaContractionOpInterface(LinalgOp linalgOp) {
+  if (!linalgOp)
+    return false;
   Operation *op = linalgOp.getOperation();
   return isa<ContractionOpInterface>(op) ||
          (isContractionInterfaceImpl(op) == MatchContractionResult::Success);