File tree Expand file tree Collapse file tree 1 file changed +5
-15
lines changed Expand file tree Collapse file tree 1 file changed +5
-15
lines changed Original file line number Diff line number Diff line change @@ -15,25 +15,15 @@ combiner {
15
15
%1 = llvm.fadd %arg0 , %arg1 : f32
16
16
omp.yield (%1 : f32 )
17
17
}
18
- atomic {
19
- ^bb2 (%arg2: !llvm.ptr , %arg3: !llvm.ptr ):
20
- %2 = llvm.load %arg3 : !llvm.ptr -> f32
21
- llvm.atomicrmw fadd %arg2 , %2 monotonic : !llvm.ptr , f32
22
- omp.yield
23
- }
24
18
25
19
llvm.func @simple_reduction (%lb : i64 , %ub : i64 , %step : i64 ) {
26
20
%c1 = llvm.mlir.constant (1 : i32 ) : i32
27
21
%0 = llvm.alloca %c1 x i32 : (i32 ) -> !llvm.ptr
28
- omp.parallel {
29
- omp.wsloop reduction (@add_f32 %0 -> %prv : !llvm.ptr )
30
- for (%iv ) : i64 = (%lb ) to (%ub ) step (%step ) {
31
- %1 = llvm.mlir.constant (2.0 : f32 ) : f32
32
- %2 = llvm.load %prv : !llvm.ptr -> f32
33
- %3 = llvm.fadd %1 , %2 : f32
34
- llvm.store %3 , %prv : f32 , !llvm.ptr
35
- omp.yield
36
- }
22
+ omp.parallel reduction (@add_f32 %0 -> %prv : !llvm.ptr ) {
23
+ %1 = llvm.mlir.constant (2.0 : f32 ) : f32
24
+ %2 = llvm.load %prv : !llvm.ptr -> f32
25
+ %3 = llvm.fadd %1 , %2 : f32
26
+ llvm.store %3 , %prv : f32 , !llvm.ptr
37
27
omp.terminator
38
28
}
39
29
llvm.return
You can’t perform that action at this time.
0 commit comments