diff options
Diffstat (limited to 'mlir/lib/Analysis/LoopAnalysis.cpp')
| -rw-r--r-- | mlir/lib/Analysis/LoopAnalysis.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/mlir/lib/Analysis/LoopAnalysis.cpp b/mlir/lib/Analysis/LoopAnalysis.cpp index 8406a37d793..de98849136c 100644 --- a/mlir/lib/Analysis/LoopAnalysis.cpp +++ b/mlir/lib/Analysis/LoopAnalysis.cpp @@ -194,7 +194,8 @@ static bool isVectorElement(LoadOrStoreOpPointer memoryOp) { // TODO(ntv): make the following into MLIR instructions, then use isa<>. static bool isVectorTransferReadOrWrite(const Statement &stmt) { const auto *opStmt = cast<OperationStmt>(&stmt); - return isaVectorTransferRead(*opStmt) || isaVectorTransferWrite(*opStmt); + return opStmt->isa<VectorTransferReadOp>() || + opStmt->isa<VectorTransferWriteOp>(); } using VectorizableStmtFun = |
