Skip to content

Commit a726997

Browse files
committed
Simplify test
1 parent 9b64ec7 commit a726997

File tree

1 file changed

+5
-15
lines changed

1 file changed

+5
-15
lines changed

mlir/test/Target/LLVMIR/openmp-reduction-call.mlir

Lines changed: 5 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -15,25 +15,15 @@ combiner {
1515
%1 = llvm.fadd %arg0, %arg1 : f32
1616
omp.yield (%1 : f32)
1717
}
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-
}
2418

2519
llvm.func @simple_reduction(%lb : i64, %ub : i64, %step : i64) {
2620
%c1 = llvm.mlir.constant(1 : i32) : i32
2721
%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
3727
omp.terminator
3828
}
3929
llvm.return

0 commit comments

Comments
 (0)