Add extra test

Xilinx · Nov 20, 2024 · ed7f19e · ed7f19e
1 parent ca055da
commit ed7f19e
Showing 1 changed file with 29 additions and 1 deletion.
diff --git a/mlir/test/Conversion/SCFToEmitC/for.mlir b/mlir/test/Conversion/SCFToEmitC/for.mlir
@@ -125,7 +125,7 @@ func.func @for_yield_index(%arg0 : index, %arg1 : index, %arg2 : index) -> index
 // CHECK:   }
 
 
- func.func @for_yield_i32(%arg0 : index, %arg1 : index, %arg2 : index) -> i32 {
+func.func @for_yield_i32(%arg0 : index, %arg1 : index, %arg2 : index) -> i32 {
    %zero = arith.constant 0 : i32
    %r = scf.for %i0 = %arg0 to %arg1 step %arg2 iter_args(%acc = %zero) -> i32 {
      scf.yield %acc : i32
@@ -145,3 +145,31 @@ func.func @for_yield_index(%arg0 : index, %arg1 : index, %arg2 : index) -> index
 // CHECK:   }
 // CHECK:   return %3 : i32
 // CHECK: }
+
+
+func.func @for_yield_update_loop_carried_var(%arg0 : index, %arg1 : index, %arg2 : index) -> index {
+   %zero = arith.constant 0 : index
+   %r = scf.for %i0 = %arg0 to %arg1 step %arg2 iter_args(%acc = %zero) -> index {
+     %sn = arith.addi %acc, %acc : index
+     scf.yield %sn: index
+   }
+   return %r : index
+ }
+
+// CHECK: func.func @for_yield_update_loop_carried_var(%arg0: index, %arg1: index, %arg2: index) -> index {
+// CHECK:   %0 = builtin.unrealized_conversion_cast %arg2 : index to !emitc.size_t
+// CHECK:   %1 = builtin.unrealized_conversion_cast %arg1 : index to !emitc.size_t
+// CHECK:   %2 = builtin.unrealized_conversion_cast %arg0 : index to !emitc.size_t
+// CHECK:   %c0 = arith.constant 0 : index
+// CHECK:   %3 = builtin.unrealized_conversion_cast %c0 : index to !emitc.size_t
+// CHECK:   %4 = "emitc.variable"() <{value = #emitc.opaque<"">}> : () -> !emitc.size_t
+// CHECK:   emitc.assign %3 : !emitc.size_t to %4 : !emitc.size_t
+// CHECK:   emitc.for %arg3 = %2 to %1 step %0 {
+// CHECK:     %6 = builtin.unrealized_conversion_cast %4 : !emitc.size_t to index
+// CHECK:     %7 = arith.addi %6, %6 : index
+// CHECK:     %8 = builtin.unrealized_conversion_cast %7 : index to !emitc.size_t
+// CHECK:     emitc.assign %8 : !emitc.size_t to %4 : !emitc.size_t
+// CHECK:   }
+// CHECK:   %5 = builtin.unrealized_conversion_cast %4 : !emitc.size_t to index
+// CHECK:   return %5 : index
+// CHECK: }