[MLIR][OpenMP] tile Construct Operation

[abidmalikwaterloo]

Summary

This patch defines the tile operation(2.11.9.1 OpenMP tile construct) for OMP Dialect.

[kiranchandramohan]

I think we should discuss this operation a bit more before proceeding. If the loop and its control are not part of the tile OpenMP dialect operation then there is no guarantee that the tiling transformation will happen. The loop can be modified or converted by other transformations or conversions.

mlir/include/mlir/Dialect/OpenMP/OpenMPOps.td

let description = [{
     The taskyield construct specifies that the current task can be suspended
     in favor of execution of a different task.
   }];
 
   let assemblyFormat = "attr-dict";
 }
 
 //===----------------------------------------------------------------------===//
+// 2.11.9.1 tile construct
+//===----------------------------------------------------------------------===//
+
+def TileOp : OpenMP_Op<"tile"> {
+  let summary = "tile construct";
+  let description = [{
+    The tile construct tiles the outer n loops of the associated loop nest, 
+    where n is the number of items in the sizes clause, which consists of 
+    items s1, . . . , sn. 
+  }];
+
+  let arguments = (ins Variadic<AnyInteger>:$tilesize);
+  
+  let regions = (region AnyRegion:$region);
+  
+  let assemblyFormat = [{ ( `size` `(` $tilesize^ `:` type($tilesize )`)` )?  
+     $region attr-dict
+  }]; 
+}
+
+//===----------------------------------------------------------------------===//
 // 2.13.7 flush Construct
 //===----------------------------------------------------------------------===//
 def FlushOp : OpenMP_Op<"flush"> {
   let summary = "flush construct";
   let description = [{
     The flush construct executes the OpenMP flush operation. This operation
     makes a thread’s temporary view of memory consistent with memory and
     enforces an order on the memory operations of the variables explicitly
     specified or implied.
   }];
 
   let arguments = (ins Variadic<OpenMP_PointerLikeType>:$varList);
 
-  let assemblyFormat = [{ ( `(` $varList^ `:` type($varList) `)` )? attr-dict}];
+  let assemblyFormat = [{( `(` $varList^ `:` type($varList) `)` )? attr-dict}];
   let extraClassDeclaration = [{
     /// The number of variable operands.
     unsigned getNumVariableOperands() {
       return getOperation()->getNumOperands();
     }
     /// The i-th variable operand passed.
     Value getVariableOperand(unsigned i) {
       return getOperand(i);
     }
   }];
 }
+
 //===----------------------------------------------------------------------===//
 // 2.14.5 target construct
 //===----------------------------------------------------------------------===//
 
 def TargetOp : OpenMP_Op<"target",[AttrSizedOperandSegments]> {
   let summary = "target construct";
   let description = [{
     The target construct includes a region of code which is to be executed
let assemblyFormat = [{
     oilist( `if` `(` $if_expr `)`
           | `device` `(` $device `:` type($device) `)`
           | `thread_limit` `(` $thread_limit `:` type($thread_limit) `)`
           | `nowait` $nowait
     ) $region attr-dict
   }];
 }
 
-
 //===----------------------------------------------------------------------===//
 // 2.16 master Construct
 //===----------------------------------------------------------------------===//
 def MasterOp : OpenMP_Op<"master"> {
   let summary = "master construct";
   let description = [{
     The master construct specifies a structured block that is executed by
     the master thread of the team.