Compiler Explorer

Source code

module convcorr_m
  implicit none

public

contains
  !*******************************************************************************
  ! 3D convolution y = y + f*x                                                   *
  ! or                                                                           *
  ! 3D correlation f = f + x'*y                                                  *
  !*******************************************************************************
  subroutine cc_sconv3d(x,y,n1,n2,n3,f,l1n,l1p,l2n,l2p,l3n,l3p,cc,reset)

implicit none

! Arguments
    integer         , intent(in   ) :: n1, n2, n3, l1n, l1p, l2n, l2p, l3n, l3p
    real            , intent(in   ) :: x(n1,n2,n3)
    real            , intent(inout) :: y(n1,n2,n3)
    real            , intent(inout) :: f(l1n:l1p,l2n:l2p,l3n:l3p)
    character(len=*), intent(in   ), optional :: cc
    logical         , intent(in   ), optional :: reset

! Local variables
    integer :: mode, i2, i3, j1, j2, j3, i1i, i1f, i2i, i2f, i3i, i3f
    real    :: sdot

!*******************************************************************************

if (present(reset)) then
          if (reset) y(:,:,:) = 0.0
       end if

do j3 = l3n, l3p
          i3i = max(1,1+j3) ; i3f = min(n3,n3+j3)

do i3 = i3i, i3f
             do j2 = l2n, l2p
                i2i = max(1,1+j2) ; i2f = min(n2,n2+j2)
                do i2 = i2i, i2f
                   do j1 = l1n, l1p
                      i1i = max(1,1+j1) ; i1f = min(n1,n1+j1)
                      !call saxpy(i1f-i1i+1,f(j1,j2,j3),x(i1i-j1,i2-j2,i3-j3),1,y(i1i,i2,i3),1)
                      associate(inn => i1f-i1i+1)
                         y(i1i:i1f,i2,i3) = y(i1i:i1f,i2,i3) + f(j1,j2,j3)*x(i1i-j1:i1i-j1+inn,i2-j2,i3-j3)
                      end associate
                   end do
                end do
             end do
          end do

end do

end subroutine cc_sconv3d

end module convcorr_m