Logo ROOT  
Reference Guide
TMVA::DNN::TDevice Class Reference

TDevice.

The TDevice class provides static functions for the generation of CUDA grids for kernel launches and is used to encapsulate the distribution of threads and blocks over the data.

Definition at line 36 of file Device.h.

Static Public Member Functions

static dim3 BlockDims1D ()
 
static dim3 BlockDims2D ()
 
template<typename AMatrix >
static dim3 GridDims1D (const AMatrix &A)
 
template<typename AMatrix >
static dim3 GridDims2D (const AMatrix &A)
 
static dim3 GridDims2D (int nrows, int ncols)
 
template<typename AMatrix >
static int NThreads (const AMatrix &A)
 

Static Public Attributes

static constexpr int BlockDimX = 1
 
static constexpr int BlockDimY = 32
 
static constexpr int BlockSize = BlockDimX * BlockDimY
 

#include <TMVA/DNN/Architectures/Cuda/Device.h>

Member Function Documentation

◆ BlockDims1D()

static dim3 TMVA::DNN::TDevice::BlockDims1D ( )
inlinestatic

Definition at line 48 of file Device.h.

◆ BlockDims2D()

static dim3 TMVA::DNN::TDevice::BlockDims2D ( )
inlinestatic

Definition at line 55 of file Device.h.

◆ GridDims1D()

template<typename AMatrix >
static dim3 TMVA::DNN::TDevice::GridDims1D ( const AMatrix &  A)
inlinestatic

Definition at line 63 of file Device.h.

◆ GridDims2D() [1/2]

template<typename AMatrix >
static dim3 TMVA::DNN::TDevice::GridDims2D ( const AMatrix &  A)
inlinestatic

Definition at line 88 of file Device.h.

◆ GridDims2D() [2/2]

static dim3 TMVA::DNN::TDevice::GridDims2D ( int  nrows,
int  ncols 
)
inlinestatic

Definition at line 74 of file Device.h.

◆ NThreads()

template<typename AMatrix >
static int TMVA::DNN::TDevice::NThreads ( const AMatrix &  A)
inlinestatic

Definition at line 101 of file Device.h.

Member Data Documentation

◆ BlockDimX

constexpr int TMVA::DNN::TDevice::BlockDimX = 1
staticconstexpr

Definition at line 40 of file Device.h.

◆ BlockDimY

constexpr int TMVA::DNN::TDevice::BlockDimY = 32
staticconstexpr

Definition at line 42 of file Device.h.

◆ BlockSize

constexpr int TMVA::DNN::TDevice::BlockSize = BlockDimX * BlockDimY
staticconstexpr

Definition at line 44 of file Device.h.


The documentation for this class was generated from the following file: