#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	nnClusterizerSizeInputRow = 3

int	nnClusterizerSizeInputPad = 3

int	nnClusterizerSizeInputTime = 3

int	nnClusterizerElementSize = -1

bool	nnClusterizerAddIndexData = true

float	nnClassThreshold = 0.16

bool	nnSigmoidTrafoClassThreshold = 1

int	nnClusterizerUseCfRegression = 0

int	nnClusterizerBatchedMode = 1

int	nnClusterizerTotalClusters = 1

int	nnClusterizerVerbosity = 0

int	nnClusterizerBoundaryFillValue = -1

int	nnClusterizerDumpDigits = 0

int	nnClusterizerApplyCfDeconvolution = 0

int	nnClusterizerModelClassNumOutputNodes = -1

int	nnClusterizerModelReg1NumOutputNodes = -1

int	nnClusterizerModelReg2NumOutputNodes = -1

int	nnClusterizerDtype = 0

int	mISector = -1

uint	class2_elements = 0

float *	inputData32 = nullptr

OrtDataType::Float16_t *	inputData16 = nullptr

float *	outputDataClass = nullptr

float *	modelProbabilities = nullptr

float *	outputDataReg1 = nullptr

float *	outputDataReg2 = nullptr

ChargePos *	peakPositions = nullptr

bool *	clusterFlags = nullptr

float *	centralCharges = 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 size_t	getAlignmentMod (size_t addr)

template<size_t alignment = GPUCA_BUFFER_ALIGNMENT>
static size_t	getAlignment (size_t addr)

template<size_t alignment = GPUCA_BUFFER_ALIGNMENT>
static size_t	nextMultipleOf (size_t size)

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 21 of file GPUTPCNNClusterizer.cxx.

◆ RegisterMemoryAllocation()

void GPUTPCNNClusterizer::RegisterMemoryAllocation ( )

Definition at line 50 of file GPUTPCNNClusterizer.cxx.

◆ setIOPointers()

void * GPUTPCNNClusterizer::setIOPointers ( void * mem )

Definition at line 25 of file GPUTPCNNClusterizer.cxx.

◆ SetMaxData()

void GPUTPCNNClusterizer::SetMaxData ( const GPUTrackingInOutPointers & io )

Definition at line 23 of file GPUTPCNNClusterizer.cxx.

Member Data Documentation

◆ centralCharges

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

Definition at line 71 of file GPUTPCNNClusterizer.h.

◆ class2_elements

uint o2::gpu::GPUTPCNNClusterizer::class2_elements = 0

Definition at line 61 of file GPUTPCNNClusterizer.h.

◆ clusterFlags

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

Definition at line 70 of file GPUTPCNNClusterizer.h.

◆ inputData16

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

Definition at line 63 of file GPUTPCNNClusterizer.h.

◆ inputData32

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

Definition at line 62 of file GPUTPCNNClusterizer.h.

◆ mISector

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

Definition at line 58 of file GPUTPCNNClusterizer.h.

◆ mMemoryId

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

Definition at line 72 of file GPUTPCNNClusterizer.h.

◆ modelProbabilities

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

Definition at line 65 of file GPUTPCNNClusterizer.h.

◆ nnClassThreshold

float o2::gpu::GPUTPCNNClusterizer::nnClassThreshold = 0.16

Definition at line 45 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerAddIndexData

bool o2::gpu::GPUTPCNNClusterizer::nnClusterizerAddIndexData = true

Definition at line 44 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerApplyCfDeconvolution

int o2::gpu::GPUTPCNNClusterizer::nnClusterizerApplyCfDeconvolution = 0

Definition at line 53 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerBatchedMode

int o2::gpu::GPUTPCNNClusterizer::nnClusterizerBatchedMode = 1

Definition at line 48 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerBoundaryFillValue

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

Definition at line 51 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerDtype

int o2::gpu::GPUTPCNNClusterizer::nnClusterizerDtype = 0

Definition at line 57 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerDumpDigits

int o2::gpu::GPUTPCNNClusterizer::nnClusterizerDumpDigits = 0

Definition at line 52 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerElementSize

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

Definition at line 43 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerModelClassNumOutputNodes

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

Definition at line 54 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerModelReg1NumOutputNodes

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

Definition at line 55 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerModelReg2NumOutputNodes

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

Definition at line 56 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerSizeInputPad

int o2::gpu::GPUTPCNNClusterizer::nnClusterizerSizeInputPad = 3

Definition at line 41 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerSizeInputRow

int o2::gpu::GPUTPCNNClusterizer::nnClusterizerSizeInputRow = 3

Definition at line 40 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerSizeInputTime

int o2::gpu::GPUTPCNNClusterizer::nnClusterizerSizeInputTime = 3

Definition at line 42 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerTotalClusters

int o2::gpu::GPUTPCNNClusterizer::nnClusterizerTotalClusters = 1

Definition at line 49 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerUseCfRegression

int o2::gpu::GPUTPCNNClusterizer::nnClusterizerUseCfRegression = 0

Definition at line 47 of file GPUTPCNNClusterizer.h.

◆ nnClusterizerVerbosity

int o2::gpu::GPUTPCNNClusterizer::nnClusterizerVerbosity = 0

Definition at line 50 of file GPUTPCNNClusterizer.h.

◆ nnSigmoidTrafoClassThreshold

bool o2::gpu::GPUTPCNNClusterizer::nnSigmoidTrafoClassThreshold = 1

Definition at line 46 of file GPUTPCNNClusterizer.h.

◆ outputDataClass

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

Definition at line 64 of file GPUTPCNNClusterizer.h.

◆ outputDataReg1

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

Definition at line 66 of file GPUTPCNNClusterizer.h.

◆ outputDataReg2

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

Definition at line 67 of file GPUTPCNNClusterizer.h.

◆ peakPositions

ChargePos* o2::gpu::GPUTPCNNClusterizer::peakPositions = 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