Project
Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | List of all members
o2::gpu::GPUTPCNNClusterizerHost Class Reference

#include <GPUTPCNNClusterizerHost.h>

Public Member Functions

 GPUTPCNNClusterizerHost ()=default
 
 GPUTPCNNClusterizerHost (const GPUSettingsProcessingNNclusterizer &settings)
 
void init (const GPUSettingsProcessingNNclusterizer &)
 
void initClusterizer (const GPUSettingsProcessingNNclusterizer &, GPUTPCNNClusterizer &)
 
void volatileOrtAllocator (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 46 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)
inline

Definition at line 50 of file GPUTPCNNClusterizerHost.h.

Member Function Documentation

◆ getMockedAllocator()

MockedOrtAllocator * GPUTPCNNClusterizerHost::getMockedAllocator ( )

Definition at line 215 of file GPUTPCNNClusterizerHost.cxx.

◆ getMockedMemoryInfo()

const OrtMemoryInfo * GPUTPCNNClusterizerHost::getMockedMemoryInfo ( )

Definition at line 210 of file GPUTPCNNClusterizerHost.cxx.

◆ init()

void GPUTPCNNClusterizerHost::init ( const GPUSettingsProcessingNNclusterizer &  settings)

Definition at line 29 of file GPUTPCNNClusterizerHost.cxx.

◆ initClusterizer()

void GPUTPCNNClusterizerHost::initClusterizer ( const GPUSettingsProcessingNNclusterizer &  settings,
GPUTPCNNClusterizer &  clustererNN 
)

Definition at line 84 of file GPUTPCNNClusterizerHost.cxx.

◆ volatileOrtAllocator()

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

Definition at line 200 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: