[Mlir-commits] [flang] [mlir] [flang][openacc][openmp] Support implicit casting on the atomic interface (PR #114390)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Sat Nov 2 12:06:24 PDT 2024
khaki3 wrote:
Another idea is to assume a converted reference type on the LHS of `atomic.read`. The verifier checks if `x`s of `read` and `write` are the same, so we cannot convert `x` only for `read`.
```
real :: n
integer :: x
!$acc atomic capture
n = x
x = n * n
!$acc end atomic
```
could become
```
%0 = fir.alloca f32 {bindc_name = "n", uniq_name = "_QFEn"}
%1:2 = hlfir.declare %0 {uniq_name = "_QFEn"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
%2 = fir.convert %1#1 : (!fir.ref<f32>) -> !fir.ref<i32>
%3 = fir.alloca i32 {bindc_name = "x", uniq_name = "_QFEx"}
%4:2 = hlfir.declare %3 {uniq_name = "_QFEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
%5 = fir.load %1#0 : !fir.ref<f32>
%6 = fir.load %1#0 : !fir.ref<f32>
%7 = arith.mulf %5, %6 fastmath<contract> : f32
%8 = fir.convert %7 : (f32) -> i32
acc.atomic.capture {
acc.atomic.read %2 = %4#1 : !fir.ref<i32>, f32
acc.atomic.write %4#1 = %8 : !fir.ref<i32>, i32
}
```
My first attempt was to extend the declaration of `read` to accept the original type of `v`. But it was a destructive change. The current change in this PR adds a temp for the `read` LHS.
https://github.com/llvm/llvm-project/pull/114390
More information about the Mlir-commits
mailing list