DiFfRG
Loading...
Searching...
No Matches
Public Types | Public Member Functions | Private Types | Private Attributes | List of all members
DiFfRG::Integrator4D2AngGPU< NT, KERNEL > Class Template Reference

GPU integrator for the integration of a 4D function with two angles with CUDA. Calculates. More...

#include <integrator_4D_2ang_gpu.hh>

Public Types

using ctype = typename get_type::ctype<NT>
 Numerical type to be used for integration tasks e.g. the argument or possible jacobians.
 

Public Member Functions

 Integrator4D2AngGPU (QuadratureProvider &quadrature_provider, const std::array< uint, 3 > grid_sizes, const ctype x_extent, const JSONValue &json)
 
 Integrator4D2AngGPU (QuadratureProvider &quadrature_provider, std::array< uint, 3 > grid_sizes, const ctype x_extent, const uint max_block_size=256)
 
 Integrator4D2AngGPU (const Integrator4D2AngGPU &other)
 
template<typename... T>
NT get (const ctype k, const T &...t) const
 Get the integral of the kernel.
 
template<typename... T>
std::future< NT > request (const ctype k, const T &...t) const
 Request a future for the integral of the kernel.
 

Private Types

using PoolMR = rmm::mr::pool_memory_resource<rmm::mr::device_memory_resource>
 

Private Attributes

QuadratureProvider & quadrature_provider
 
const std::array< uint, 3 > grid_sizes
 
std::array< uint, 3 > block_sizes
 
const uint device_data_size
 
const ctype * ptr_x_quadrature_p
 
const ctype * ptr_x_quadrature_w
 
const ctype * ptr_ang_quadrature_p
 
const ctype * ptr_ang_quadrature_w
 
const ctype x_extent
 
dim3 num_blocks
 
dim3 threads_per_block
 
int n_devices
 
std::vector< std::shared_ptr< PoolMR > > pool
 
const rmm::cuda_stream_pool cuda_stream_pool
 
std::atomic_ullong evaluations
 

Detailed Description

template<typename NT, typename KERNEL>
class DiFfRG::Integrator4D2AngGPU< NT, KERNEL >

GPU integrator for the integration of a 4D function with two angles with CUDA. Calculates.

\[ 2\pi\,\int dp\, d\text{cos}_1\, d\text{cos}_2\, \frac{1}{(2\pi)^4} \sqrt{1-\text{cos}_1^2} f(p, \text{cos}_1, \text{cos}_2, ...) + c \]

with \( p^2 \) bounded by \( \text{x_extent} * k^2 \).

Template Parameters
NTThe numerical type of the result.
KERNELThe kernel to integrate.

Member Typedef Documentation

◆ ctype

template<typename NT , typename KERNEL >
using DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::ctype = typename get_type::ctype<NT>

Numerical type to be used for integration tasks e.g. the argument or possible jacobians.

◆ PoolMR

template<typename NT , typename KERNEL >
using DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::PoolMR = rmm::mr::pool_memory_resource<rmm::mr::device_memory_resource>
private

Constructor & Destructor Documentation

◆ Integrator4D2AngGPU() [1/3]

template<typename NT , typename KERNEL >
DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::Integrator4D2AngGPU ( QuadratureProvider & quadrature_provider,
const std::array< uint, 3 > grid_sizes,
const ctype x_extent,
const JSONValue & json )
inline

◆ Integrator4D2AngGPU() [2/3]

template<typename NT , typename KERNEL >
DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::Integrator4D2AngGPU ( QuadratureProvider & quadrature_provider,
std::array< uint, 3 > grid_sizes,
const ctype x_extent,
const uint max_block_size = 256 )
inline

◆ Integrator4D2AngGPU() [3/3]

template<typename NT , typename KERNEL >
DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::Integrator4D2AngGPU ( const Integrator4D2AngGPU< NT, KERNEL > & other)
inline

Member Function Documentation

◆ get()

template<typename NT , typename KERNEL >
template<typename... T>
NT DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::get ( const ctype k,
const T &... t ) const
inline

Get the integral of the kernel.

Template Parameters
TTypes of the parameters for the kernel.
Parameters
kRG-scale.
tParameters forwarded to the kernel.
Returns
NT Integral of the kernel plus the constant part.

◆ request()

template<typename NT , typename KERNEL >
template<typename... T>
std::future< NT > DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::request ( const ctype k,
const T &... t ) const
inline

Request a future for the integral of the kernel.

Template Parameters
TTypes of the parameters for the kernel.
Parameters
kRG-scale.
tParameters forwarded to the kernel.
Returns
std::future<NT> future holding the integral of the kernel plus the constant part.

Member Data Documentation

◆ block_sizes

template<typename NT , typename KERNEL >
std::array<uint, 3> DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::block_sizes
private

◆ cuda_stream_pool

template<typename NT , typename KERNEL >
const rmm::cuda_stream_pool DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::cuda_stream_pool
private

◆ device_data_size

template<typename NT , typename KERNEL >
const uint DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::device_data_size
private

◆ evaluations

template<typename NT , typename KERNEL >
std::atomic_ullong DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::evaluations
mutableprivate

◆ grid_sizes

template<typename NT , typename KERNEL >
const std::array<uint, 3> DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::grid_sizes
private

◆ n_devices

template<typename NT , typename KERNEL >
int DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::n_devices
private

◆ num_blocks

template<typename NT , typename KERNEL >
dim3 DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::num_blocks
private

◆ pool

template<typename NT , typename KERNEL >
std::vector<std::shared_ptr<PoolMR> > DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::pool
mutableprivate

◆ ptr_ang_quadrature_p

template<typename NT , typename KERNEL >
const ctype* DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::ptr_ang_quadrature_p
private

◆ ptr_ang_quadrature_w

template<typename NT , typename KERNEL >
const ctype* DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::ptr_ang_quadrature_w
private

◆ ptr_x_quadrature_p

template<typename NT , typename KERNEL >
const ctype* DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::ptr_x_quadrature_p
private

◆ ptr_x_quadrature_w

template<typename NT , typename KERNEL >
const ctype* DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::ptr_x_quadrature_w
private

◆ quadrature_provider

template<typename NT , typename KERNEL >
QuadratureProvider& DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::quadrature_provider
private

◆ threads_per_block

template<typename NT , typename KERNEL >
dim3 DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::threads_per_block
private

◆ x_extent

template<typename NT , typename KERNEL >
const ctype DiFfRG::Integrator4D2AngGPU< NT, KERNEL >::x_extent
private
The documentation for this class was generated from the following file: