mesh_batch.F90

Go to the documentation of this file.

!! Copyright (C) 2002-2006 M. Marques, A. Castro, A. Rubio, G. Bertsch
!!
!! This program is free software; you can redistribute it and/or modify
!! it under the terms of the GNU General Public License as published by
!! the Free Software Foundation; either version 2, or (at your option)
!! any later version.
!!
!! This program is distributed in the hope that it will be useful,
!! but WITHOUT ANY WARRANTY; without even the implied warranty of
!! MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
!! GNU General Public License for more details.
!!
!! You should have received a copy of the GNU General Public License
!! along with this program; if not, write to the Free Software
!! Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
!! 02110-1301, USA.
!!
 
#include "global.h"
 
module mesh_batch_oct_m
  use accel_oct_m
  use accel_blas_oct_m
  use batch_oct_m
  use batch_ops_oct_m
  use blas_oct_m
  use iso_c_binding
  use debug_oct_m
  use global_oct_m
  use hardware_oct_m
  use lalg_basic_oct_m
  use math_oct_m
  use mesh_oct_m
  use mesh_function_oct_m
  use messages_oct_m
  use mpi_oct_m
  use mpi_debug_oct_m
  use multicomm_oct_m
#if defined(HAVE_OPENMP)
  use omp_lib
#endif
  use par_vec_oct_m
  use partition_oct_m
  use profiling_oct_m
  use types_oct_m
 
  implicit none
 
  private
  public ::                         &
    batch_p_t,                      &
    dmesh_batch_dotp_matrix,        &
    zmesh_batch_dotp_matrix,        &
    dmesh_batch_dotp_vector,        &
    zmesh_batch_dotp_vector,        &
    dmesh_batch_dotp_self,          &
    zmesh_batch_dotp_self,          &
    dmesh_batch_exchange_points,    &
    zmesh_batch_exchange_points,    &
    mesh_batch_nrm2,                &
    dmesh_batch_orthogonalization,  &
    zmesh_batch_orthogonalization,  &
    dmesh_batch_mf_dotp,            &
    zmesh_batch_mf_dotp,            &
    dmesh_batch_codensity,          &
    zmesh_batch_codensity,          &
    smesh_batch_codensity,          &
    cmesh_batch_codensity,          &
    dmesh_batch_normalize,          &
    zmesh_batch_normalize,          &
    mesh_batch_weight_device
 
  type batch_p_t
    class(batch_t), allocatable :: p
  end type batch_p_t
 
 
contains
 
  ! -----------------------------------------------------
  !
  subroutine mesh_batch_nrm2(mesh, aa, nrm2, reduce)
    class(mesh_t),      intent(in)    :: mesh
    class(batch_t),     intent(in)    :: aa
    real(real64), contiguous,  intent(out)   :: nrm2(:)
    logical, optional,  intent(in)    :: reduce
 
    push_sub(mesh_batch_nrm2)
 
    if (aa%type() == type_float) then
      call dpriv_mesh_batch_nrm2(mesh, aa, nrm2)
    else
      call zpriv_mesh_batch_nrm2(mesh, aa, nrm2)
    end if
 
    if (mesh%parallel_in_domains .and. optional_default(reduce, .true.)) then
      nrm2(1:aa%nst) = nrm2(1:aa%nst)**2
      call mesh%allreduce(nrm2, dim = aa%nst)
      nrm2(1:aa%nst) = sqrt(nrm2(1:aa%nst))
    end if
 
    pop_sub(mesh_batch_nrm2)
  end subroutine mesh_batch_nrm2
 
  ! -----------------------------------------------------
  subroutine mesh_batch_weight_device(mesh, aa, tmp, sqrt_weights)
    class(mesh_t),               intent(in)  :: mesh
    class(batch_t),              intent(in)  :: aa
    class(batch_t), allocatable, intent(out) :: tmp
    logical, optional,           intent(in)  :: sqrt_weights
 
    type(accel_kernel_t), save :: kernel_scale
    integer(int64), dimension(3) :: gsizes, bsizes
 
    push_sub(mesh_batch_weight_device)
 
    assert(mesh%use_vol_pp)
    assert(aa%status() == batch_device_packed)
 
    call aa%clone_to(tmp)
    assert(tmp%status() == batch_device_packed)
 
    call accel_kernel_start_call(kernel_scale, 'mesh_batch_single.cu', 'scale_points')
 
    call accel_set_kernel_arg(kernel_scale, 0, mesh%np)
    call accel_set_kernel_arg(kernel_scale, 1, int(aa%pack_size_real(1), int32))
    call accel_set_kernel_arg(kernel_scale, 2, int(aa%pack_size_real(1), int32))
    call accel_set_kernel_arg(kernel_scale, 3, int(tmp%pack_size_real(1), int32))
    call accel_set_kernel_arg(kernel_scale, 4, 0)
    if (optional_default(sqrt_weights, .false.)) then
      call accel_set_kernel_arg(kernel_scale, 5, mesh%sqrt_vol_pp_buffer)
    else
      call accel_set_kernel_arg(kernel_scale, 5, mesh%vol_pp_buffer)
    end if
    call accel_set_kernel_arg(kernel_scale, 6, aa%ff_device)
    call accel_set_kernel_arg(kernel_scale, 7, tmp%ff_device)
 
    call accel_grid_size_extend_dim(int(mesh%np, int64), aa%pack_size_real(1), gsizes, bsizes, kernel_scale)
 
    call accel_kernel_run(kernel_scale, gsizes, bsizes)
 
    pop_sub(mesh_batch_weight_device)
  end subroutine mesh_batch_weight_device
 
#undef SPECIALIZED
#include "undef.F90"
#include "real.F90"
#include "mesh_batch_inc.F90"
 
#include "undef.F90"
#include "complex.F90"
#include "mesh_batch_inc.F90"
 
! Specialized kernels for nst_linear=2
#define SPECIALIZED 2
#include "undef.F90"
#include "real.F90"
#include "mesh_batch_inc.F90"
 
#include "undef.F90"
#include "complex.F90"
#include "mesh_batch_inc.F90"
#undef SPECIALIZED
 
! Specialized kernels for nst_linear=3
#define SPECIALIZED 3
#include "undef.F90"
#include "real.F90"
#include "mesh_batch_inc.F90"
 
#include "undef.F90"
#include "complex.F90"
#include "mesh_batch_inc.F90"
#undef SPECIALIZED
 
! Specialized kernels for nst_linear=4
#define SPECIALIZED 4
#include "undef.F90"
#include "real.F90"
#include "mesh_batch_inc.F90"
 
#include "undef.F90"
#include "complex.F90"
#include "mesh_batch_inc.F90"
#undef SPECIALIZED
 
! Specialized kernels for nst_linear=6
#define SPECIALIZED 6
#include "undef.F90"
#include "real.F90"
#include "mesh_batch_inc.F90"
 
#include "undef.F90"
#include "complex.F90"
#include "mesh_batch_inc.F90"
#undef SPECIALIZED
 
! Specialized kernels for nst_linear=8
#define SPECIALIZED 8
#include "undef.F90"
#include "real.F90"
#include "mesh_batch_inc.F90"
 
#include "undef.F90"
#include "complex.F90"
#include "mesh_batch_inc.F90"
#undef SPECIALIZED
 
! Co-density kernel, instantiated for all precisions (d, z, s, c)
#include "undef.F90"
#include "real.F90"
#include "mesh_batch_codensity_inc.F90"
 
#include "undef.F90"
#include "complex.F90"
#include "mesh_batch_codensity_inc.F90"
 
#include "undef.F90"
#include "real32.F90"
#include "mesh_batch_codensity_inc.F90"
 
#include "undef.F90"
#include "complex32.F90"
#include "mesh_batch_codensity_inc.F90"
 
end module mesh_batch_oct_m
 
!! Local Variables:
!! mode: f90
!! coding: utf-8
!! End: