sum_3d_tensor_int64 Function

private function sum_3d_tensor_int64(tensor, threaded) result(res)

perform the total sum of all components of type int64 in a 3d tensor

Arguments

Type IntentOptional Attributes Name
integer(kind=int64), intent(in) :: tensor(:,:,:)
logical, intent(in), optional :: threaded

Return Value integer(kind=int64)


Calls

proc~~sum_3d_tensor_int64~~CallsGraph proc~sum_3d_tensor_int64 sum_3d_tensor_int64 interface~pic_optional pic_optional proc~sum_3d_tensor_int64->interface~pic_optional proc~optional_char optional_char interface~pic_optional->proc~optional_char proc~optional_dp optional_dp interface~pic_optional->proc~optional_dp proc~optional_int32 optional_int32 interface~pic_optional->proc~optional_int32 proc~optional_int64 optional_int64 interface~pic_optional->proc~optional_int64 proc~optional_logical optional_logical interface~pic_optional->proc~optional_logical proc~optional_sp optional_sp interface~pic_optional->proc~optional_sp

Called by

proc~~sum_3d_tensor_int64~~CalledByGraph proc~sum_3d_tensor_int64 sum_3d_tensor_int64 interface~pic_sum pic_sum interface~pic_sum->proc~sum_3d_tensor_int64

Variables

Type Visibility Attributes Name Initial
integer(kind=default_int), private :: i
integer(kind=default_int), private :: ii
integer(kind=default_int), private :: j
integer(kind=default_int), private :: jj
integer(kind=default_int), private :: k
integer(kind=default_int), private :: kk
integer(kind=default_int), private :: nx
integer(kind=default_int), private :: ny
integer(kind=default_int), private :: nz
logical, private :: use_threads

Source Code

   function sum_3d_tensor_int64(tensor, threaded) result(res)
    !! perform the total sum of all components of type int64 in a 3d tensor
      integer(int64), intent(in) :: tensor(:, :, :)
      logical, intent(in), optional :: threaded
      logical :: use_threads
      integer(int64) :: res
      integer(default_int) :: nx, ny, nz
      integer(default_int) :: i, j, k, ii, jj, kk

      nx = size(tensor, 1)
      ny = size(tensor, 2)
      nz = size(tensor, 3)

      use_threads = pic_optional(threaded, use_threaded_default)
      res = 0_int64

      if (use_threads) then
         !$omp parallel do collapse(3) private(i,j,k,ii,jj,kk) reduction(+: res)
         do kk = 1, nz, block_size
            do jj = 1, ny, block_size
               do ii = 1, nx, block_size
                  do k = kk, min(kk + block_size - 1, nz)
                     do j = jj, min(jj + block_size - 1, ny)
                        do i = ii, min(ii + block_size - 1, nx)
                           res = res + tensor(i, j, k)
                        end do
                     end do
                  end do
               end do
            end do
         end do
         !$omp end parallel do
      else
         res = sum(tensor)
      end if

   end function sum_3d_tensor_int64