Add linalg.transpose

mlir/dialects/linalg.py

@@ -323,6 +323,20 @@ class LinalgMatvec(DialectOp):
                  " outs( {c_id.ssa_id} : {c_type.type} )")]
 
 
+@dataclass
+class LinalgTranspose(DialectOp):
+    inarg: List[mast.SsaId]
+    in_type: List[mast.Type]
+    init: List[mast.SsaId]
+    init_type: List[mast.Type]
+    permutation: List[int]
+
+    _syntax_ = [("linalg.transpose"
+                 " ins( {inarg.ssa_id_list} : {in_type.type_list_no_parens} )"
+                 " outs( {init.ssa_id_list} : {init_type.type_list_no_parens} )"
+                 " permutation = [ {permutation.ssa_use_list} ]")]
+
+
 # Inspect current module to get all classes defined above
 linalg = Dialect("linalg", ops=[m[1] for m in inspect.getmembers(
     sys.modules[__name__], lambda obj: is_op(obj, __name__))])

tests/test_linalg.py

@@ -167,6 +167,15 @@ def test_matvec():
 }""")
 
 
+def test_transpose():
+    assert_roundtrip_equivalence("""module {
+  func.func @transpose(%arg0: memref<?x?xf16>, %arg1: memref<?x?xf16>) {
+    %transpose = linalg.transpose ins( %arg0 : memref<?x?xf16> ) outs( %arg1 : memref<?x?xf16> ) permutation = [ 1, 0 ]
+    return
+  }
+}""")
+
+
 if __name__ == "__main__":
     if len(sys.argv) > 1:
         exec(sys.argv[1])