#include <GPUTPCNNClusterizer.h>

Inherits o2::gpu::GPUProcessor.

Public Member Functions
	GPUTPCNNClusterizer ()=default

void *	setIOPointers (void *)

void	RegisterMemoryAllocation ()

void	InitializeProcessor ()

void	SetMaxData (const GPUTrackingInOutPointers &)

Public Member Functions inherited from o2::gpu::GPUProcessor
	GPUProcessor ()

	~GPUProcessor ()

	GPUProcessor (const GPUProcessor &)=delete

GPUProcessor &	operator= (const GPUProcessor &)=delete

	GPUd () GPUconstantref() const GPUConstantMem *GetConstantMem() const

	GPUd () GPUconstantref() const GPUParam &Param() const

	GPUd () void raiseError(uint32_t code

const GPUReconstruction &	GetRec () 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 void *	alignPointer (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 S *	getPointerWithAlignment (size_t &basePtr, size_t nEntries=1)

template<size_t alignment = GPUCA_BUFFER_ALIGNMENT, class S >
static S *	getPointerWithAlignment (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
GPUReconstruction *	mRec

ProcessorType	mGPUProcessorType

GPUProcessor *	mLinkedProcessor

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:

/home/runner/work/AliceO2/AliceO2/GPU/GPUTracking/TPCClusterFinder/GPUTPCNNClusterizer.h
/home/runner/work/AliceO2/AliceO2/GPU/GPUTracking/TPCClusterFinder/GPUTPCNNClusterizer.cxx

Public Member Functions

Public Attributes

Additional Inherited Members