[mlir][openacc] Restrict types for acc.bounds operands to index or integer type

[clementval]

Summary

Values in bounds are expected to have integer or index types. Enforce
this expectation by restricting the type to be IntOrIndex.

[razvanlupusoru]

Thank you! This constraint makes sense.

mlir/include/mlir/Dialect/OpenACC/OpenACCOps.td

 }
 
 // Used for data specification in data clauses (2.7.1).
 // Either (or both) extent and upperbound must be specified.
 def OpenACC_DataBoundsOp : OpenACC_Op<"bounds",
     [AttrSizedOperandSegments, NoMemoryEffect]> {
   let summary = "Represents bounds information for acc data clause.";
 
-  let arguments = (ins Optional<AnyType>:$lowerbound,
-                       Optional<AnyType>:$upperbound,
-                       Optional<AnyType>:$extent,
-                       Optional<AnyType>:$stride,
+  let arguments = (ins Optional<IntOrIndex>:$lowerbound,
+                       Optional<IntOrIndex>:$upperbound,
+                       Optional<IntOrIndex>:$extent,
+                       Optional<IntOrIndex>:$stride,
                        DefaultValuedAttr<BoolAttr, "false">:$strideInBytes,
-                       Optional<AnyType>:$startIdx);
+                       Optional<IntOrIndex>:$startIdx);
   let results = (outs OpenACC_DataBoundsType:$result);
 
   let assemblyFormat = [{
     oilist(
         `lowerbound` `(` $lowerbound `:` type($lowerbound) `)`
       | `upperbound` `(` $upperbound `:` type($upperbound) `)`
       | `extent` `(` $extent `:` type($extent) `)`
       | `stride` `(` $stride `:` type($stride) `)`

mlir/test/Dialect/OpenACC/invalid.mlir

 acc.enter_data wait(%cst: index) create(%value : memref<10xf32>) attributes {wait}
 
 // -----
 
 %cst = arith.constant 1 : index
 %value = memref.alloc() : memref<10xf32>
 // expected-error@+1 {{wait_devnum cannot appear without waitOperands}}
 acc.enter_data wait_devnum(%cst: index) create(%value : memref<10xf32>)
+
+// -----
+
+%0 = arith.constant 1.0 : f32
+// expected-error@+1 {{operand #0 must be integer or index, but got 'f32'}}
+%1 = acc.bounds lowerbound(%0 : f32)
+