@@ -79,7 +79,7 @@ def ForOp : Loop_Op<"for",
79
79
%sum_0 = constant 0.0 : f32
80
80
// iter_args binds initial values to the loop's region arguments.
81
81
%sum = loop.for %iv = %lb to %ub step %step iter_args(%sum_iter = %sum_0) -> (f32) {
82
- %t = load %buffer[%iv] : memref<1024xf32>
82
+ %t = load %buffer[%iv] : memref<1024xf32>
83
83
%sum_next = addf %sum_iter, %t : f32
84
84
// Yield current iteration sum to next iteration %sum_iter or to %sum if final iteration.
85
85
loop.yield %sum_next : f32
@@ -100,14 +100,14 @@ def ForOp : Loop_Op<"for",
100
100
%sum_0 = constant 0.0 : f32
101
101
%c0 = constant 0.0 : f32
102
102
%sum = loop.for %iv = %lb to %ub step %step iter_args(%sum_iter = %sum_0) -> (f32) {
103
- %t = load %buffer[%iv] : memref<1024xf32>
104
- %cond = cmpf "ugt", %t, %c0 : f32
105
- %sum_next = loop.if %cond -> (f32) {
106
- %new_sum = addf %sum_iter, %t : f32
103
+ %t = load %buffer[%iv] : memref<1024xf32>
104
+ %cond = cmpf "ugt", %t, %c0 : f32
105
+ %sum_next = loop.if %cond -> (f32) {
106
+ %new_sum = addf %sum_iter, %t : f32
107
107
loop.yield %new_sum : f32
108
- } else {
109
- loop.yield %sum_iter : f32
110
- }
108
+ } else {
109
+ loop.yield %sum_iter : f32
110
+ }
111
111
loop.yield %sum_next : f32
112
112
}
113
113
return %sum : f32
0 commit comments