[Mlir-commits] [mlir] 0f8d40f - [mlir][linalg] Add a builder for `linalg.(indexed_)generic`.

[Alexander Belyaev]

Author: Alexander Belyaev
Date: 2020-06-10T23:52:41+02:00
New Revision: 0f8d40f0d655db49bddca6d9c1cb3d3577219e6c

URL: https://github.com/llvm/llvm-project/commit/0f8d40f0d655db49bddca6d9c1cb3d3577219e6c
DIFF: https://github.com/llvm/llvm-project/commit/0f8d40f0d655db49bddca6d9c1cb3d3577219e6c.diff

LOG: [mlir][linalg] Add a builder for `linalg.(indexed_)generic`.

Differential Revision: https://reviews.llvm.org/D81610

Added: 
    

Modified: 
    mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td b/mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td
index c13a79087d85..cc7eb5ce9d68 100644
--- a/mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td
+++ b/mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td
@@ -565,6 +565,21 @@ class GenericOpBase<string mnemonic> : LinalgStructuredBase_Op<mnemonic,
         "No such thing as reference indexing maps for a generic op.");
      }
   }];
+
+  let builders = [
+    OpBuilder<"OpBuilder &builder, OperationState &result, "
+              "ArrayRef<Type> resultTypes, ValueRange args, "
+              "int64_t inputCount, int64_t outputCount, "
+              "ArrayRef<AffineMap> indexingMaps, "
+              "ArrayRef<StringRef> iteratorTypes", [{
+        return build(builder, result, resultTypes, args,
+                     builder.getI64IntegerAttr(inputCount),
+                     builder.getI64IntegerAttr(outputCount),
+                     builder.getAffineMapArrayAttr(indexingMaps),
+                     builder.getStrArrayAttr(iteratorTypes),
+                     /*doc=*/nullptr, /*library_call=*/nullptr);
+  }]>];
+
   let printer = [{ return ::print(p, *this); }];
   let parser = [{ return ::parseGenericOp(parser, result); }];
 }