[mlir][linalg][transform][python] Add type arg to MatchOp extension.
The extension class to MatchOp has a class method called match_op_names. The previous version of that function did not allow to specify the result type. This, however, may be useful/necessary if the op consuming the resulting handle requires a particular type (such as the bufferization.EmptyTensorToAllocTensorOp). This patch adds an overload to match_op_names that allows to specify the result type. Reviewed By: ftynse Differential Revision: https://reviews.llvm.org/D155567
Loading
Please sign in to comment