o2::gpu::GPUTPCNNClusterizerHost Class Reference

#include <GPUTPCNNClusterizerHost.h>

Public Member Functions
	GPUTPCNNClusterizerHost ()=default
	GPUTPCNNClusterizerHost (const GPUSettingsProcessingNNclusterizer &settings, bool useDeterministicMode=false)
void	init (const GPUSettingsProcessingNNclusterizer &, bool=false)
void	initClusterizer (const GPUSettingsProcessingNNclusterizer &, GPUTPCNNClusterizer &, int32_t=-1, int32_t=-1)
void	createBoundary (GPUTPCNNClusterizer &)
void	createIndexLookup (GPUTPCNNClusterizer &)
void	directOrtAllocator (Ort::Env *, Ort::MemoryInfo *, GPUReconstruction *, bool=false)
MockedOrtAllocator *	getMockedAllocator ()
const OrtMemoryInfo *	getMockedMemoryInfo ()

Public Attributes
std::unordered_map< std::string, std::string >	mOrtOptions
o2::ml::OrtModel	mModelClass
o2::ml::OrtModel	mModelReg1
o2::ml::OrtModel	mModelReg2
std::vector< bool >	mModelsUsed = {false, false, false}
int32_t	mDeviceId = -1
std::shared_ptr< MockedOrtAllocator >	mMockedAlloc = nullptr

Detailed Description

Definition at line 44 of file GPUTPCNNClusterizerHost.h.

Constructor & Destructor Documentation

GPUTPCNNClusterizerHost() [1/2]

o2::gpu::GPUTPCNNClusterizerHost::GPUTPCNNClusterizerHost ( )

default

GPUTPCNNClusterizerHost() [2/2]

o2::gpu::GPUTPCNNClusterizerHost::GPUTPCNNClusterizerHost ( const GPUSettingsProcessingNNclusterizer &  settings, bool  useDeterministicMode = false  )

inline

Definition at line 48 of file GPUTPCNNClusterizerHost.h.

Member Function Documentation

createBoundary()

void o2::gpu::GPUTPCNNClusterizerHost::createBoundary

(

GPUTPCNNClusterizer &

)

createIndexLookup()

void o2::gpu::GPUTPCNNClusterizerHost::createIndexLookup

(

GPUTPCNNClusterizer &

)

directOrtAllocator()

void GPUTPCNNClusterizerHost::directOrtAllocator	(	Ort::Env *	env,
		Ort::MemoryInfo *	memInfo,
		GPUReconstruction *	rec,
		bool	recreate = false
	)

Definition at line 252 of file GPUTPCNNClusterizerHost.cxx.

getMockedAllocator()

MockedOrtAllocator * GPUTPCNNClusterizerHost::getMockedAllocator

(

)

Definition at line 267 of file GPUTPCNNClusterizerHost.cxx.

getMockedMemoryInfo()

const OrtMemoryInfo * GPUTPCNNClusterizerHost::getMockedMemoryInfo

(

)

Definition at line 262 of file GPUTPCNNClusterizerHost.cxx.

init()

void GPUTPCNNClusterizerHost::init	(	const GPUSettingsProcessingNNclusterizer &	settings,
		bool	useDeterministicMode = false
	)

Definition at line 32 of file GPUTPCNNClusterizerHost.cxx.

initClusterizer()

void GPUTPCNNClusterizerHost::initClusterizer	(	const GPUSettingsProcessingNNclusterizer &	settings,
		GPUTPCNNClusterizer &	clustererNN,
		int32_t	maxFragmentLen = -1,
		int32_t	maxAllowedTimebin = -1
	)

Definition at line 88 of file GPUTPCNNClusterizerHost.cxx.

Member Data Documentation

mDeviceId

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

Definition at line 63 of file GPUTPCNNClusterizerHost.h.

mMockedAlloc

std::shared_ptr<MockedOrtAllocator> o2::gpu::GPUTPCNNClusterizerHost::mMockedAlloc = nullptr

Definition at line 64 of file GPUTPCNNClusterizerHost.h.

mModelClass

o2::ml::OrtModel o2::gpu::GPUTPCNNClusterizerHost::mModelClass

Definition at line 61 of file GPUTPCNNClusterizerHost.h.

mModelReg1

o2::ml::OrtModel o2::gpu::GPUTPCNNClusterizerHost::mModelReg1

Definition at line 61 of file GPUTPCNNClusterizerHost.h.

mModelReg2

o2::ml::OrtModel o2::gpu::GPUTPCNNClusterizerHost::mModelReg2

Definition at line 61 of file GPUTPCNNClusterizerHost.h.

mModelsUsed

std::vector<bool> o2::gpu::GPUTPCNNClusterizerHost::mModelsUsed = {false, false, false}

Definition at line 62 of file GPUTPCNNClusterizerHost.h.

mOrtOptions

std::unordered_map<std::string, std::string> o2::gpu::GPUTPCNNClusterizerHost::mOrtOptions

Definition at line 60 of file GPUTPCNNClusterizerHost.h.

---

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

• /home/runner/work/AliceO2/AliceO2/GPU/GPUTracking/TPCClusterFinder/GPUTPCNNClusterizerHost.h
• /home/runner/work/AliceO2/AliceO2/GPU/GPUTracking/TPCClusterFinder/GPUTPCNNClusterizerHost.cxx