fill_3d_tensor_sp Subroutine

private subroutine fill_3d_tensor_sp(tensor, alpha, threaded)

fill a tensor of datatype sp with a value

Arguments

Type IntentOptional Attributes Name
real(kind=sp), intent(inout) :: tensor(:,:,:)
real(kind=sp), intent(in) :: alpha
logical, intent(in), optional :: threaded

Calls

proc~~fill_3d_tensor_sp~~CallsGraph proc~fill_3d_tensor_sp fill_3d_tensor_sp interface~pic_optional pic_optional proc~fill_3d_tensor_sp->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~~fill_3d_tensor_sp~~CalledByGraph proc~fill_3d_tensor_sp fill_3d_tensor_sp interface~pic_fill pic_fill interface~pic_fill->proc~fill_3d_tensor_sp

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

   subroutine fill_3d_tensor_sp(tensor, alpha, threaded)
     !! fill a tensor of datatype sp with a value
      real(sp), intent(inout) :: tensor(:, :, :)
      real(sp), intent(in)    :: alpha
      integer(default_int) :: i, j, k
      integer(default_int) :: ii, jj, kk
      integer(default_int) :: nx, ny, nz
      logical, intent(in), optional :: threaded
      logical :: use_threads

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

      use_threads = pic_optional(threaded, use_threaded_default)

      if (use_threads) then
         !$omp parallel do collapse(3) private(i,j,k,ii,jj,kk)
         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)
                           tensor(i, j, k) = alpha
                        end do
                     end do
                  end do
               end do
            end do
         end do
         !$omp end parallel do
      else
         tensor = alpha
      end if

   end subroutine fill_3d_tensor_sp