maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

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

This commit is contained in:
Nicolas Vasilache 2021-02-05 13:37:00 +00:00
parent 1b101038dc
commit 6da8d6c68f
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlir/lib/Dialect/Linalg/IR/LinalgInterfaces.cpp
View File

@ -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);