Fortran: simplification of FINDLOC for constant complex arguments [PR110585]

gcc/fortran/ChangeLog:

	PR fortran/110585
	* arith.cc (gfc_compare_expr): Handle equality comparison of constant
	complex gfc_expr arguments.

gcc/testsuite/ChangeLog:

	PR fortran/110585
	* gfortran.dg/findloc_9.f90: New test.

(cherry picked from commit 7ac1581d066a6f3a0d4acf1042a74634258b4966)
This commit is contained in:
Harald Anlauf 2023-07-07 20:25:06 +02:00
parent 238944752e
commit d00a7d22c6
2 changed files with 24 additions and 0 deletions

View File

@ -1120,6 +1120,11 @@ gfc_compare_expr (gfc_expr *op1, gfc_expr *op2, gfc_intrinsic_op op)
|| (op1->value.logical && !op2->value.logical));
break;
case BT_COMPLEX:
gcc_assert (op == INTRINSIC_EQ);
rc = mpc_cmp (op1->value.complex, op2->value.complex);
break;
default:
gfc_internal_error ("gfc_compare_expr(): Bad basic type");
}

View File

@ -0,0 +1,19 @@
! { dg-do compile }
! { dg-options "-fdump-tree-original" }
! PR fortran/110585 - simplification of FINDLOC for constant complex arguments
program mvce
implicit none
integer, parameter :: a(*) = findloc([(1.,0.),(2.,1.)], (2.,0.))
integer, parameter :: b(*) = findloc([(1.,0.),(2.,1.)], (2.,0.), back=.true.)
integer, parameter :: c(*) = findloc([(1.,0.),(2.,1.)], (2.,1.))
integer, parameter :: d(*) = findloc([(1.,0.),(2.,1.)], (2.,1.), back=.true.)
integer, parameter :: e = findloc([(1.,0.),(2.,1.)], (2.,1.), dim=1)
if (a(1) /= 0) stop 1
if (b(1) /= 0) stop 2
if (c(1) /= 2) stop 3
if (d(1) /= 2) stop 4
if (e /= 2) stop 5
end
! { dg-final { scan-tree-dump-not "_gfortran_stop_numeric" "original" } }