Skip to content

Commit 095070f

Browse files
author
Jerry Wu
committed
Add res() method
1 parent 7df28fd commit 095070f

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

mlir/include/mlir/Dialect/Linalg/IR/LinalgInterfaces.td

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,14 @@ def LinalgContractionOpInterface : OpInterface<"ContractionOpInterface"> {
5454
return $_op.getOperation()->getOperand(1);
5555
}]>,
5656
InterfaceMethod<
57+
/*desc=*/"Returns the result value.",
58+
/*retTy=*/"OpResult",
59+
/*methodName=*/"res",
60+
/*args=*/(ins),
61+
/*methodBody=*/[{
62+
return $_op.getOperation()->getResult(1);
63+
}]>,
64+
InterfaceMethod<
5765
/*desc=*/[{
5866
Returns whether the given op has indexing maps that correspond to a
5967
row-major matmul operation.

0 commit comments

Comments
 (0)