[Flang] Segmentation fault in an assignment statement as an array with a subscript in FORALL construct

[ohno-fj]

Version of flang-new : 17.0.0(f6c4808d95221a5838e14474d95c6fe85bb1488a)
AArch64

An assignment statement using an array as an array index in the FORALL construct results in a segmentation fault.
Changing the array index from v (i) to i, as shown below, result in successful compilation.

• Before the change

forall (i = 1: 5)
v (i) = i
b (v (i): v (i)) = mod 01 _ fun1 (a (v (i): v (i)))
end forall

• After the change

forall (i = 1: 5)
b (i: i) = mod01 _ fun1 (a (i: i))
end forall

The following are the test program, results of Flang-new, GFortran and ifort compilation and execution.

snf95_120_2.f90:

program main
  call test01()
  print *,'pass'
end program main

module mod01
contains
  elemental function mod01_fun1(i) result(ifun)
    intent(in) :: i
    ifun=i
  end function mod01_fun1
end module mod01

subroutine test01()
  use mod01
  integer v(5),a(5),b(5)
  a=reshape((/(i,i=1,5)/),(/5/))
  b=0
  write(6,*) "a = ", a
  forall (i=1:5)
     v(i)=i
     b(v(i):v(i))=mod01_fun1(a(v(i):v(i)))
  end forall
  write(6,*) "b = ", b
  if (any(a.ne.b)) write(6,*) "NG"
end subroutine test01

$ flang-new -flang-experimental-exec snf95_120_2.f90; ./a.out
 a =  1 2 3 4 5
Segmentation fault (core dumped)
$

$ gfortran snf95_120_2.f90; ./a.out
 a =            1           2           3           4           5
 b =            1           2           3           4           5
 pass
$

$ ifort snf95_120_2.f90; ./a.out
 a =            1           2           3           4           5
 b =            1           2           3           4           5
 pass
$

[llvmbot]

@llvm/issue-subscribers-flang-frontend

[llvmbot]

@llvm/issue-subscribers-bug

[luporl]

https://reviews.llvm.org/D153454

[llvmbot]

@llvm/issue-subscribers-flang-ir