diff options
| author | Nicolas Vasilache <ntv@google.com> | 2019-12-11 09:26:51 -0800 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-12-11 09:27:34 -0800 |
| commit | 508d4e672e5de9b9c582fc5e96996d93bb92c74e (patch) | |
| tree | 267dac5084c2c7e12bc56db7de79cc423ebefe50 /mlir/test/lib | |
| parent | c5fb4c1303837a41eb06b14137d3e4a5387023a3 (diff) | |
| download | bcm5719-llvm-508d4e672e5de9b9c582fc5e96996d93bb92c74e.tar.gz bcm5719-llvm-508d4e672e5de9b9c582fc5e96996d93bb92c74e.zip | |
Continue refactoring StructuredOps utilities
This CL adds more common information to StructuredOpsUtils.h
The n_view attribute is retired in favor of args_in + args_out but the CL is otherwise NFC.
PiperOrigin-RevId: 285000621
Diffstat (limited to 'mlir/test/lib')
| -rw-r--r-- | mlir/test/lib/DeclarativeTransforms/TestLinalgTransformPatterns.td | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/mlir/test/lib/DeclarativeTransforms/TestLinalgTransformPatterns.td b/mlir/test/lib/DeclarativeTransforms/TestLinalgTransformPatterns.td index 9ba2fc4a734..4d8c9282f2d 100644 --- a/mlir/test/lib/DeclarativeTransforms/TestLinalgTransformPatterns.td +++ b/mlir/test/lib/DeclarativeTransforms/TestLinalgTransformPatterns.td @@ -108,7 +108,7 @@ def : Pattern<(DotOp:$op $a, $b, $c), //===----------------------------------------------------------------------===// // Linalg to vector contraction patterns. //===----------------------------------------------------------------------===// -def : Pattern<(GenericOp:$op $_1, $_2, $_3, $_4, $_5, $_6, $_7), +def : Pattern<(GenericOp:$op $_1, $_2, $_3, $_4, $_5, $_6, $_7, $_8), [(LinalgOpToVectorContraction<"GenericOp"> $op)], [(Constraint<HasLinalgTransformMarker<"_marked_matmul_">> $op)]>; @@ -116,13 +116,14 @@ def : Pattern<(GenericOp:$op $_1, $_2, $_3, $_4, $_5, $_6, $_7), // Linalg generic permutation patterns. //===----------------------------------------------------------------------===// -def : Pat<(GenericOp:$op $input, $imap, $itypes, $nviews, $doc, $fun, $libcall), +def : Pat<(GenericOp:$op $_1, $_2, $_3, $_4, $_5, $_6, $_7, $_8), (PermuteGenericLinalgOp<[1,2,0],"PERMUTED"> $op), [(Constraint<And<[HasNoLinalgTransformMarker, AffineMapDomainHasDim<3>]>> $op)]>; -def : Pat<(IndexedGenericOp:$op $input, $imap, $itypes, $nviews, $doc, $fun, $libcall), +def : Pat<(IndexedGenericOp:$op $_1, $_2, $_3, $_4, $_5, $_6, $_7, $_8), (PermuteGenericLinalgOp<[1,2,0],"PERMUTED"> $op), [(Constraint<And<[HasNoLinalgTransformMarker, AffineMapDomainHasDim<3>]>> $op)]>; + #endif // TEST_LINALG_TRANSFORMS_PATTERNS |
