[mlir][affine][test] Check if affine.for has bounding maps composed

[eopXD]

Summary

This is test for bug listed on https://bugs.llvm.org/show_bug.cgi?id=45203

[eopXD]

This commit shall be added under https://reviews.llvm.org/D99323

mlir/test/Dialect/Affine/canonicalize.mlir

func @compose_affine_maps_div_symbol(%A : memref<i64>, %i0 : index, %i1 : index) {
   store %3, %A[]: memref<i64>
   affine.for %i2 = 0 to 3 {
     %4 = affine.apply affine_map<(d0)[s0, s1] -> (d0 ceildiv s1 + s0 + s0 * 3)> (%i2)[%0, %1]
     %5 = index_cast %4: index to i64
     store %5, %A[]: memref<i64>
   }
   return
 }
+
+// -----
+
+// Ensure bounding maps of affine.for is composed
+
+// CHECK: #map = affine_map<()[s0] -> (s0 + 2)>
+// CHECK-NEXT: module {
+// CHECK-NEXT:   func @compose_affine_for_bounds(%arg0: index) {
+// CHECK-NEXT:     affine.for %arg1 = 0 to #map()[%arg0] {
+// CHECK-NEXT:     "foo"() : () -> ()
+// CHECK-NEXT:   }
+// CHECK-NEXT:   return 
+// CHECK-NEXT: }
+
+func @compose_affine_for_bounds(%arg0: index) {
+  %v = affine.apply affine_map<(d0) -> (d0 + 2)>(%arg0)
+  affine.for %i = 0 to %v {
+    "foo"() : () -> ()
+  }
+  return
+}
 No newline at end of file