#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
int	mNnClusterizerSizeInputRow = 3

int	mNnClusterizerSizeInputPad = 3

int	mNnClusterizerSizeInputTime = 3

int	mNnClusterizerElementSize = -1

bool	mNnClusterizerAddIndexData = true

float	mNnClassThreshold = 0.01

bool	mNnSigmoidTrafoClassThreshold = 1

int	mNnClusterizerUseCfRegression = 0

int	mNnClusterizerBatchedMode = 1

int	mNnClusterizerTotalClusters = 1

int	mNnClusterizerVerbosity = 0

int	mNnClusterizerBoundaryFillValue = -1

int	mNnClusterizerModelClassNumOutputNodes = -1

int	mNnClusterizerModelReg1NumOutputNodes = -1

int	mNnClusterizerModelReg2NumOutputNodes = -1

int	mNnInferenceInputDType = 0

int	mNnInferenceOutputDType = 0

int	mISector = -1

int	mDeviceId = -1

bool *	mClusterFlags = nullptr

int *	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 95 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

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

Definition at line 62 of file GPUTPCNNClusterizer.h.

◆ mDeviceId

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

Definition at line 58 of file GPUTPCNNClusterizer.h.

◆ mInputData_16

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

Definition at line 72 of file GPUTPCNNClusterizer.h.

◆ mInputData_32

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

Definition at line 66 of file GPUTPCNNClusterizer.h.

◆ mISector

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

Definition at line 57 of file GPUTPCNNClusterizer.h.

◆ mMemoryId

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

Definition at line 77 of file GPUTPCNNClusterizer.h.

◆ mModelProbabilities_16

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

Definition at line 73 of file GPUTPCNNClusterizer.h.

◆ mModelProbabilities_32

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

Definition at line 67 of file GPUTPCNNClusterizer.h.

◆ mNnClassThreshold

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

Definition at line 45 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerAddIndexData

bool o2::gpu::GPUTPCNNClusterizer::mNnClusterizerAddIndexData = true

Definition at line 44 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerBatchedMode

int o2::gpu::GPUTPCNNClusterizer::mNnClusterizerBatchedMode = 1

Definition at line 48 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerBoundaryFillValue

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

Definition at line 51 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerElementSize

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

Definition at line 43 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerModelClassNumOutputNodes

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

Definition at line 52 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerModelReg1NumOutputNodes

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

Definition at line 53 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerModelReg2NumOutputNodes

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

Definition at line 54 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerSizeInputPad

int o2::gpu::GPUTPCNNClusterizer::mNnClusterizerSizeInputPad = 3

Definition at line 41 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerSizeInputRow

int o2::gpu::GPUTPCNNClusterizer::mNnClusterizerSizeInputRow = 3

Definition at line 40 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerSizeInputTime

int o2::gpu::GPUTPCNNClusterizer::mNnClusterizerSizeInputTime = 3

Definition at line 42 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerTotalClusters

int o2::gpu::GPUTPCNNClusterizer::mNnClusterizerTotalClusters = 1

Definition at line 49 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerUseCfRegression

int o2::gpu::GPUTPCNNClusterizer::mNnClusterizerUseCfRegression = 0

Definition at line 47 of file GPUTPCNNClusterizer.h.

◆ mNnClusterizerVerbosity

int o2::gpu::GPUTPCNNClusterizer::mNnClusterizerVerbosity = 0

Definition at line 50 of file GPUTPCNNClusterizer.h.

◆ mNnInferenceInputDType

int o2::gpu::GPUTPCNNClusterizer::mNnInferenceInputDType = 0

Definition at line 55 of file GPUTPCNNClusterizer.h.

◆ mNnInferenceOutputDType

int o2::gpu::GPUTPCNNClusterizer::mNnInferenceOutputDType = 0

Definition at line 56 of file GPUTPCNNClusterizer.h.

◆ mNnSigmoidTrafoClassThreshold

bool o2::gpu::GPUTPCNNClusterizer::mNnSigmoidTrafoClassThreshold = 1

Definition at line 46 of file GPUTPCNNClusterizer.h.

◆ mOutputDataClass

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

Definition at line 63 of file GPUTPCNNClusterizer.h.

◆ mOutputDataReg1_16

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

Definition at line 74 of file GPUTPCNNClusterizer.h.

◆ mOutputDataReg1_32

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

Definition at line 68 of file GPUTPCNNClusterizer.h.

◆ mOutputDataReg2_16

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

Definition at line 75 of file GPUTPCNNClusterizer.h.

◆ mOutputDataReg2_32

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

Definition at line 69 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