69b3303e6b
This patch adds lowering for some numeric related intrinsics: - `exponent` - `floor` - `fraction` - `mod` - `modulo` - `nint` - `not` - `product` This patch is part of the upstreaming effort from fir-dev branch. Reviewed By: PeteSteinfeld, schweitz Differential Revision: https://reviews.llvm.org/D121828 Co-authored-by: Peter Steinfeld <psteinfeld@nvidia.com> Co-authored-by: Jean Perier <jperier@nvidia.com> Co-authored-by: mleair <leairmark@gmail.com> Co-authored-by: Eric Schweitz <eschweitz@nvidia.com>
38 lines
2.1 KiB
Fortran
38 lines
2.1 KiB
Fortran
! RUN: bbc -emit-fir %s -o - | FileCheck %s
|
|
|
|
! CHECK-LABEL: func @_QPmodulo_testr(
|
|
! CHECK-SAME: %[[arg0:.*]]: !fir.ref<f64>{{.*}}, %[[arg1:.*]]: !fir.ref<f64>{{.*}}, %[[arg2:.*]]: !fir.ref<f64>{{.*}}) {
|
|
subroutine modulo_testr(r, a, p)
|
|
real(8) :: r, a, p
|
|
! CHECK-DAG: %[[a:.*]] = fir.load %[[arg1]] : !fir.ref<f64>
|
|
! CHECK-DAG: %[[p:.*]] = fir.load %[[arg2]] : !fir.ref<f64>
|
|
! CHECK-DAG: %[[rem:.*]] = arith.remf %[[a]], %[[p]] : f64
|
|
! CHECK-DAG: %[[zero:.*]] = arith.constant 0.000000e+00 : f64
|
|
! CHECK-DAG: %[[remNotZero:.*]] = arith.cmpf une, %[[rem]], %[[zero]] : f64
|
|
! CHECK-DAG: %[[aNeg:.*]] = arith.cmpf olt, %[[a]], %[[zero]] : f64
|
|
! CHECK-DAG: %[[pNeg:.*]] = arith.cmpf olt, %[[p]], %[[zero]] : f64
|
|
! CHECK-DAG: %[[signDifferent:.*]] = arith.xori %[[aNeg]], %[[pNeg]] : i1
|
|
! CHECK-DAG: %[[mustAddP:.*]] = arith.andi %[[remNotZero]], %[[signDifferent]] : i1
|
|
! CHECK-DAG: %[[remPlusP:.*]] = arith.addf %[[rem]], %[[p]] : f64
|
|
! CHECK: %[[res:.*]] = arith.select %[[mustAddP]], %[[remPlusP]], %[[rem]] : f64
|
|
! CHECK: fir.store %[[res]] to %[[arg0]] : !fir.ref<f64>
|
|
r = modulo(a, p)
|
|
end subroutine
|
|
|
|
! CHECK-LABEL: func @_QPmodulo_testi(
|
|
! CHECK-SAME: %[[arg0:.*]]: !fir.ref<i64>{{.*}}, %[[arg1:.*]]: !fir.ref<i64>{{.*}}, %[[arg2:.*]]: !fir.ref<i64>{{.*}}) {
|
|
subroutine modulo_testi(r, a, p)
|
|
integer(8) :: r, a, p
|
|
! CHECK-DAG: %[[a:.*]] = fir.load %[[arg1]] : !fir.ref<i64>
|
|
! CHECK-DAG: %[[p:.*]] = fir.load %[[arg2]] : !fir.ref<i64>
|
|
! CHECK-DAG: %[[rem:.*]] = arith.remsi %[[a]], %[[p]] : i64
|
|
! CHECK-DAG: %[[argXor:.*]] = arith.xori %[[a]], %[[p]] : i64
|
|
! CHECK-DAG: %[[signDifferent:.*]] = arith.cmpi slt, %[[argXor]], %c0{{.*}} : i64
|
|
! CHECK-DAG: %[[remNotZero:.*]] = arith.cmpi ne, %[[rem]], %c0{{.*}} : i64
|
|
! CHECK-DAG: %[[mustAddP:.*]] = arith.andi %[[remNotZero]], %[[signDifferent]] : i1
|
|
! CHECK-DAG: %[[remPlusP:.*]] = arith.addi %[[rem]], %[[p]] : i64
|
|
! CHECK: %[[res:.*]] = arith.select %[[mustAddP]], %[[remPlusP]], %[[rem]] : i64
|
|
! CHECK: fir.store %[[res]] to %[[arg0]] : !fir.ref<i64>
|
|
r = modulo(a, p)
|
|
end subroutine
|
|
|