Replace dyn_casts with isa_and_present

Xilinx · Aug 7, 2024 · 369a1e2 · 369a1e2
1 parent eccc213
commit 369a1e2
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/mlir/lib/Conversion/AIRToAIEPass.cpp b/mlir/lib/Conversion/AIRToAIEPass.cpp
@@ -3429,13 +3429,13 @@ class AIRLinalgOpToLibraryCallRewrite
     auto getLibFnOperands = [](linalg::LinalgOp op) {
       SmallVector<Value> operands;
       for (auto operand : op->getOperands()) {
-        if (auto operation = operand.getDefiningOp()) {
-          if (dyn_cast<memref::ReshapeOp>(operation) ||
-              dyn_cast<memref::ExpandShapeOp>(operation) ||
-              dyn_cast<memref::CollapseShapeOp>(operation)) {
+        auto operation = operand.getDefiningOp();
+        if (isa_and_present<
+                    memref::ReshapeOp, 
+                    memref::ExpandShapeOp, 
+                    memref::CollapseShapeOp>(operation)) {
             operands.push_back(operation->getOperand(0));
             continue;
-          }
         }
         operands.push_back(operand);
       }