[MLIR][Shape] Add convenience builder for `shape.assuming_all`

[frgossen]

Summary

Depends On D100104

[bkramer]

mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td

751

Maybe add a comment that we can't generate it automatically because the op has variadic arguments.

This is a bug in our tblgen emitter, but I don't know what it takes to fix it.

[frgossen]

Address comment

[frgossen]

mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td

751

Shortly looked into this and the case (which is explicitly skipped).
There is a TODO for generalization though.
Will try to generalize this if it's quick enough.

mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td

let description = [{
     %wf = shape.assuming_all %w0, %w1 // Failure
     %wt = shape.assuming_all %w0, %w2 // Passing
     ```
   }];
 
   let arguments = (ins Variadic<Shape_WitnessType>:$inputs);
   let results = (outs Shape_WitnessType:$result);
 
+  // Only needed while tablegen is unable to generate this for ops with variadic
+  // arguments.
+  let builders = [
+    OpBuilder<(ins "ValueRange":$inputs)>,
+  ];
+
   let assemblyFormat = "$inputs attr-dict";
 
   let hasFolder = 1;
   let hasCanonicalizer = 1;
 
   let verifier = [{ return ::verify(*this); }];
 }
 

mlir/lib/Dialect/Shape/IR/Shape.cpp

 static LogicalResult verify(AssumingAllOp op) {
   // Ensure that AssumingAllOp contains at least one operand
   if (op.getNumOperands() == 0)
     return op.emitOpError("no operands specified");
 
   return success();
 }
 
+void AssumingAllOp::build(OpBuilder &b, OperationState &state,
+                          ValueRange inputs) {
+  build(b, state, b.getType<WitnessType>(), inputs);
+}
+
 //===----------------------------------------------------------------------===//
 // BroadcastOp
 //===----------------------------------------------------------------------===//
 
 OpFoldResult BroadcastOp::fold(ArrayRef<Attribute> operands) {
   if (operands.size() == 1)
     return shapes().front();