Resolve vector notation issues (encountering Fortran intrinsics)

[MichaelSt98]

What happened?

As an example, execute:

from loki import Subroutine, fgen, resolve_vector_notation

fcode_kernel = """
SUBROUTINE kernel()
INTEGER :: ARR_LEN = 3
INTEGER :: A(ARR_LEN, ARR_LEN + 1)
INTEGER :: B(ARR_LEN + 1)
B(:) = SUM(A(:,:), 2)
END SUBROUTINE kernel
"""

kernel = Subroutine.from_source(fcode_kernel)
resolve_vector_notation(kernel)
print(fgen(kernel))

which yields

SUBROUTINE kernel ()
  INTEGER :: ARR_LEN = 3
  INTEGER :: A(ARR_LEN, ARR_LEN + 1)
  INTEGER :: B(ARR_LEN + 1)
  INTEGER :: i_B_1
  INTEGER :: i_B_0
  DO i_B_1=1,ARR_LEN + 1
    DO i_B_0=1,ARR_LEN
      B(i_B_0) = SUM(A(i_B_0, i_B_1), 2)
    END DO
  END DO
END SUBROUTINE kernel

with the problematic line: B(i_B_0) = SUM(A(i_B_0, i_B_1), 2).

This is just one example of a larger problem/issue and also applies to SCCBaseTransformation.resolve_vector_dimension.

What are the steps to reproduce the bug?

Execute:

from loki import Subroutine, fgen, resolve_vector_notation

fcode_kernel = """
SUBROUTINE kernel()
INTEGER :: ARR_LEN = 3
INTEGER :: A(ARR_LEN, ARR_LEN + 1)
INTEGER :: B(ARR_LEN + 1)
B(:) = SUM(A(:,:), 2)
END SUBROUTINE kernel
"""

kernel = Subroutine.from_source(fcode_kernel)
resolve_vector_notation(kernel)
print(fgen(kernel))

Version

main

Platform (OS and architecture)

Mac / HPC2020

Relevant log output

No response

Accompanying data

No response

Organisation

No response