Project
Loading...
Searching...
No Matches
o2::gpu::GPUTPCNNClusterizer Class Reference

#include <GPUTPCNNClusterizer.h>

Inherits o2::gpu::GPUProcessor.

Public Member Functions

 GPUTPCNNClusterizer ()=default
 
voidsetIOPointers (void *)
 
void RegisterMemoryAllocation ()
 
void InitializeProcessor ()
 
void SetMaxData (const GPUTrackingInOutPointers &)
 
- Public Member Functions inherited from o2::gpu::GPUProcessor
 GPUProcessor ()
 
 ~GPUProcessor ()
 
 GPUProcessor (const GPUProcessor &)=delete
 
GPUProcessoroperator= (const GPUProcessor &)=delete
 
 GPUd () GPUconstantref() const GPUConstantMem *GetConstantMem() const
 
 GPUd () GPUconstantref() const GPUParam &Param() const
 
 GPUd () void raiseError(uint32_t code
 
const GPUReconstructionGetRec () const
 
void InitGPUProcessor (GPUReconstruction *rec, ProcessorType type=PROCESSOR_TYPE_CPU, GPUProcessor *slaveProcessor=nullptr)
 
void Clear ()
 
template<class T >
T & HostProcessor (T *)
 

Public Attributes

int32_t mNnClusterizerSizeInputRow = 3
 
int32_t mNnClusterizerSizeInputPad = 3
 
int32_t mNnClusterizerSizeInputTime = 3
 
int32_t mNnClusterizerChargeArraySize = -1
 
int32_t mNnClusterizerElementSize = -1
 
int8_t mNnClusterizerAddIndexData = 1
 
float mNnClassThreshold = 0.01
 
int8_t mNnSigmoidTrafoClassThreshold = 1
 
int8_t mNnClusterizerSetDeconvolutionFlags = 1
 
int32_t mNnClusterizerUseCfRegression = 0
 
int32_t mNnClusterizerBatchedMode = 1
 
int32_t mNnClusterizerTotalClusters = 1
 
int32_t mNnClusterizerVerbosity = 0
 
int32_t mNnClusterizerBoundaryFillValue = -1
 
int32_t mNnClusterizerModelClassNumOutputNodes = -1
 
int32_t mNnClusterizerModelReg1NumOutputNodes = -1
 
int32_t mNnClusterizerModelReg2NumOutputNodes = -1
 
int32_t mNnInferenceInputDType = 0
 
int32_t mNnInferenceOutputDType = 0
 
int32_t mISector = -1
 
int32_t mDeviceId = -1
 
uint32_t mNnClusterizerFullRowSize = 0
 
uint32_t mNnClusterizerFullPadSize = 0
 
uint32_t mNnClusterizerFullTimeSize = 0
 
uint32_t mNnClusterizerPadTimeSize = 0
 
uint32_t mNnClusterizerRowTimeSize = 0
 
uint32_t mNnClusterizerRowTimeSizeFull = 0
 
int8_t * mClusterFlags = nullptr
 
int32_t * mOutputDataClass = nullptr
 
float * mInputData_32 = nullptr
 
float * mModelProbabilities_32 = nullptr
 
float * mOutputDataReg1_32 = nullptr
 
float * mOutputDataReg2_32 = nullptr
 
OrtDataType::Float16_t * mInputData_16 = nullptr
 
OrtDataType::Float16_t * mModelProbabilities_16 = nullptr
 
OrtDataType::Float16_t * mOutputDataReg1_16 = nullptr
 
OrtDataType::Float16_t * mOutputDataReg2_16 = nullptr
 
int16_t mMemoryId = -1
 
- Public Attributes inherited from o2::gpu::GPUProcessor
uint32_t param1 = 0
 
uint32_t uint32_t param2 = 0
 
uint32_t uint32_t uint32_t param3 = 0) const
 

Additional Inherited Members

- Public Types inherited from o2::gpu::GPUProcessor
enum  ProcessorType { PROCESSOR_TYPE_CPU = 0 , PROCESSOR_TYPE_DEVICE = 1 , PROCESSOR_TYPE_SLAVE = 2 }
 
- Static Public Member Functions inherited from o2::gpu::GPUProcessor
template<size_t alignment = GPUCA_BUFFER_ALIGNMENT>
static constexpr size_t getAlignmentMod (size_t addr)
 
template<size_t alignment = GPUCA_BUFFER_ALIGNMENT>
static constexpr size_t getAlignment (size_t addr)
 
template<size_t alignment = GPUCA_BUFFER_ALIGNMENT>
static constexpr size_t nextMultipleOf (size_t size)
 
static constexpr size_t nextMultipleOf (size_t size, size_t alignment)
 
template<size_t alignment = GPUCA_BUFFER_ALIGNMENT>
static voidalignPointer (void *ptr)
 
template<size_t alignment = GPUCA_BUFFER_ALIGNMENT>
static size_t getAlignmentMod (void *addr)
 
template<size_t alignment = GPUCA_BUFFER_ALIGNMENT>
static size_t getAlignment (void *addr)
 
template<size_t alignment = GPUCA_BUFFER_ALIGNMENT, class S >
static SgetPointerWithAlignment (size_t &basePtr, size_t nEntries=1)
 
template<size_t alignment = GPUCA_BUFFER_ALIGNMENT, class S >
static SgetPointerWithAlignment (void *&basePtr, size_t nEntries=1)
 
template<size_t alignment = GPUCA_BUFFER_ALIGNMENT, class T , class S >
static void computePointerWithAlignment (T *&basePtr, S *&objPtr, size_t nEntries=1)
 
template<class T , class S >
static void computePointerWithoutAlignment (T *&basePtr, S *&objPtr, size_t nEntries=1)
 
- Protected Member Functions inherited from o2::gpu::GPUProcessor
void AllocateAndInitializeLate ()
 
 GPUconstantref () const GPUConstantMem *mConstantMem
 
- Protected Attributes inherited from o2::gpu::GPUProcessor
GPUReconstructionmRec
 
ProcessorType mGPUProcessorType
 
GPUProcessormLinkedProcessor
 

Detailed Description

Definition at line 29 of file GPUTPCNNClusterizer.h.

Constructor & Destructor Documentation

◆ GPUTPCNNClusterizer()

o2::gpu::GPUTPCNNClusterizer::GPUTPCNNClusterizer ( )
default

Member Function Documentation

◆ InitializeProcessor()

void GPUTPCNNClusterizer::InitializeProcessor ( )

Definition at line 22 of file GPUTPCNNClusterizer.cxx.

◆ RegisterMemoryAllocation()

void GPUTPCNNClusterizer::RegisterMemoryAllocation ( )

Definition at line 68 of file GPUTPCNNClusterizer.cxx.

◆ setIOPointers()

void * GPUTPCNNClusterizer::setIOPointers ( void mem)

Definition at line 26 of file GPUTPCNNClusterizer.cxx.

◆ SetMaxData()

void GPUTPCNNClusterizer::SetMaxData ( const GPUTrackingInOutPointers io)

Definition at line 24 of file GPUTPCNNClusterizer.cxx.

Member Data Documentation

◆ mClusterFlags

int8_t* o2::gpu::GPUTPCNNClusterizer::mClusterFlags = nullptr

Definition at line 83 of file GPUTPCNNClusterizer.h.

◆ mDeviceId

int32_t o2::gpu::GPUTPCNNClusterizer::mDeviceId = -1

Definition at line 60 of file GPUTPCNNClusterizer.h.

◆ mInputData_16

OrtDataType::Float16_t* o2::gpu::GPUTPCNNClusterizer::mInputData_16 = nullptr

Definition at line 93 of file GPUTPCNNClusterizer.h.

◆ mInputData_32

float* o2::gpu::GPUTPCNNClusterizer::mInputData_32 = nullptr

Definition at line 87 of file GPUTPCNNClusterizer.h.

◆ mISector

int32_t o2::gpu::GPUTPCNNClusterizer::mISector = -1

Definition at line 59 of file GPUTPCNNClusterizer.h.

◆ mMemoryId

int16_t o2::gpu::GPUTPCNNClusterizer::mMemoryId = -1

Definition at line 98 of file GPUTPCNNClusterizer.h.

◆ mModelProbabilities_16

OrtDataType::Float16_t* o2::gpu::GPUTPCNNClusterizer::mModelProbabilities_16 = nullptr

Definition at line 94 of file GPUTPCNNClusterizer.h.

◆ mModelProbabilities_32

float* o2::gpu::GPUTPCNNClusterizer::mModelProbabilities_32 = nullptr

Definition at line 88 of file GPUTPCNNClusterizer.h.

◆ mNnClassThreshold

float o2::gpu::GPUTPCNNClusterizer::mNnClassThreshold = 0.01

Definition at line 46 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerAddIndexData

int8_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerAddIndexData = 1

Definition at line 45 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerBatchedMode

int32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerBatchedMode = 1

Definition at line 50 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerBoundaryFillValue

int32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerBoundaryFillValue = -1

Definition at line 53 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerChargeArraySize

int32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerChargeArraySize = -1

Definition at line 43 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerElementSize

int32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerElementSize = -1

Definition at line 44 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerFullPadSize

uint32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerFullPadSize = 0

Definition at line 64 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerFullRowSize

uint32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerFullRowSize = 0

Definition at line 63 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerFullTimeSize

uint32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerFullTimeSize = 0

Definition at line 65 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerModelClassNumOutputNodes

int32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerModelClassNumOutputNodes = -1

Definition at line 54 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerModelReg1NumOutputNodes

int32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerModelReg1NumOutputNodes = -1

Definition at line 55 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerModelReg2NumOutputNodes

int32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerModelReg2NumOutputNodes = -1

Definition at line 56 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerPadTimeSize

uint32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerPadTimeSize = 0

Definition at line 66 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerRowTimeSize

uint32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerRowTimeSize = 0

Definition at line 67 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerRowTimeSizeFull

uint32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerRowTimeSizeFull = 0

Definition at line 68 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerSetDeconvolutionFlags

int8_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerSetDeconvolutionFlags = 1

Definition at line 48 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerSizeInputPad

int32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerSizeInputPad = 3

Definition at line 41 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerSizeInputRow

int32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerSizeInputRow = 3

Definition at line 40 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerSizeInputTime

int32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerSizeInputTime = 3

Definition at line 42 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerTotalClusters

int32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerTotalClusters = 1

Definition at line 51 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerUseCfRegression

int32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerUseCfRegression = 0

Definition at line 49 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerVerbosity

int32_t o2::gpu::GPUTPCNNClusterizer::mNnClusterizerVerbosity = 0

Definition at line 52 of file GPUTPCNNClusterizer.h.

◆ mNnInferenceInputDType

int32_t o2::gpu::GPUTPCNNClusterizer::mNnInferenceInputDType = 0

Definition at line 57 of file GPUTPCNNClusterizer.h.

◆ mNnInferenceOutputDType

int32_t o2::gpu::GPUTPCNNClusterizer::mNnInferenceOutputDType = 0

Definition at line 58 of file GPUTPCNNClusterizer.h.

◆ mNnSigmoidTrafoClassThreshold

int8_t o2::gpu::GPUTPCNNClusterizer::mNnSigmoidTrafoClassThreshold = 1

Definition at line 47 of file GPUTPCNNClusterizer.h.

◆ mOutputDataClass

int32_t* o2::gpu::GPUTPCNNClusterizer::mOutputDataClass = nullptr

Definition at line 84 of file GPUTPCNNClusterizer.h.

◆ mOutputDataReg1_16

OrtDataType::Float16_t* o2::gpu::GPUTPCNNClusterizer::mOutputDataReg1_16 = nullptr

Definition at line 95 of file GPUTPCNNClusterizer.h.

◆ mOutputDataReg1_32

float* o2::gpu::GPUTPCNNClusterizer::mOutputDataReg1_32 = nullptr

Definition at line 89 of file GPUTPCNNClusterizer.h.

◆ mOutputDataReg2_16

OrtDataType::Float16_t* o2::gpu::GPUTPCNNClusterizer::mOutputDataReg2_16 = nullptr

Definition at line 96 of file GPUTPCNNClusterizer.h.

◆ mOutputDataReg2_32

float* o2::gpu::GPUTPCNNClusterizer::mOutputDataReg2_32 = nullptr

Definition at line 90 of file GPUTPCNNClusterizer.h.


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