AthOnnx::OnnxRuntimeSvc Class Reference

Service implementing AthOnnx::IOnnxRuntimeSvc. More...

#include <OnnxRuntimeSvc.h>

Inheritance diagram for AthOnnx::OnnxRuntimeSvc:

Collaboration diagram for AthOnnx::OnnxRuntimeSvc:

Public Member Functions
virtual void	print () const
	Print the state of the service. More...
template<typename T >
void	declareServiceInterface ()
	add the given interface to the list of interfaces More...
virtual StatusCode	queryInterface (const InterfaceID &riid, void **ppvi)
	query interface for gaudi More...
MsgStream &	msg () const
MsgStream &	msg (const MSG::Level lvl) const
bool	msgLvl (const MSG::Level lvl) const
	DeclareInterfaceID (AthOnnx::IOnnxRuntimeSvc, 1, 0)
	Declare the interface that this class provides. More...
Function(s) inherited from @c Service
	OnnxRuntimeSvc (const std::string &name, ISvcLocator *svc)
virtual StatusCode	initialize () override
	Function initialising the service. More...
virtual StatusCode	finalize () override
	Function finalising the service. More...
Function(s) inherited from @c AthOnnx::IOnnxRuntimeSvc
virtual Ort::Env &	env () const override
	Return the Onnx Runtime environment object. More...

Private Attributes
std::unique_ptr< Ort::Env >	m_env
	Global runtime environment for Onnx Runtime. More...
std::vector< std::pair< const InterfaceID &(*)(), void *(*)(AsgService *)> >	m_interfaces
	list of interfaces we have More...

Detailed Description

Service implementing AthOnnx::IOnnxRuntimeSvc.

This is a very simple implementation, just managing the lifetime of some Onnx Runtime C++ objects.

Author

Attila Krasznahorkay Attil.nosp@m.a.Kr.nosp@m.aszna.nosp@m.hork.nosp@m.ay@ce.nosp@m.rn.c.nosp@m.h

Definition at line 27 of file OnnxRuntimeSvc.h.

Constructor & Destructor Documentation

OnnxRuntimeSvc()

AthOnnx::OnnxRuntimeSvc::OnnxRuntimeSvc	(	const std::string &	name,
		ISvcLocator *	svc
	)

Definition at line 8 of file OnnxRuntimeSvc.cxx.

                                                                        :
      asg::AsgService(name, svc)
   {
     declareServiceInterface<AthOnnx::IOnnxRuntimeSvc>();
   }

Member Function Documentation

DeclareInterfaceID()

AthOnnx::IOnnxRuntimeSvc::DeclareInterfaceID ( AthOnnx::IOnnxRuntimeSvc  , 1  , 0    )

inherited

Declare the interface that this class provides.

declareServiceInterface()

template<typename T >

void asg::AsgService::declareServiceInterface

inherited

add the given interface to the list of interfaces

Definition at line 76 of file AsgService.h.

  {
#ifndef XAOD_STANDALONE
    m_interfaces.emplace_back (T::interfaceID, [] (AsgService *self) -> void* {return dynamic_cast<T*>(self);});
#endif
  }

env()

Ort::Env & AthOnnx::OnnxRuntimeSvc::env ( ) const

overridevirtual

Return the Onnx Runtime environment object.

Implements AthOnnx::IOnnxRuntimeSvc.

Definition at line 38 of file OnnxRuntimeSvc.cxx.

                                     {
 
      return *m_env;
   }

finalize()

StatusCode AthOnnx::OnnxRuntimeSvc::finalize ( )

overridevirtual

Function finalising the service.

Reimplemented from asg::AsgService.

Definition at line 28 of file OnnxRuntimeSvc.cxx.

                                       {
 
      // Dekete the environment object.
      m_env.reset();
      ATH_MSG_DEBUG( "Ort::Env object deleted" );
 
      // Return gracefully.
      return StatusCode::SUCCESS;
   }

initialize()

StatusCode AthOnnx::OnnxRuntimeSvc::initialize ( )

overridevirtual

Function initialising the service.

Reimplemented from asg::AsgService.

Definition at line 13 of file OnnxRuntimeSvc.cxx.

                                         {
 
      // Create the environment object.
      Ort::ThreadingOptions tp_options;
      tp_options.SetGlobalIntraOpNumThreads(1);
      tp_options.SetGlobalInterOpNumThreads(1);
 
      m_env = std::make_unique< Ort::Env >(
            tp_options, ORT_LOGGING_LEVEL_WARNING, name().c_str());
      ATH_MSG_DEBUG( "Ort::Env object created" );
 
      // Return gracefully.
      return StatusCode::SUCCESS;
   }

msg() [1/2]

MsgStream& AthCommonMsg< Service >::msg ( ) const

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                {
    return this->msgStream();
  }

msg() [2/2]

MsgStream& AthCommonMsg< Service >::msg ( const MSG::Level  lvl ) const

inlineinherited

Definition at line 27 of file AthCommonMsg.h.

                                                  {
    return this->msgStream(lvl);
  }

msgLvl()

bool AthCommonMsg< Service >::msgLvl ( const MSG::Level  lvl ) const

inlineinherited

Definition at line 30 of file AthCommonMsg.h.

                                               {
    return this->msgLevel(lvl);
  }

print()

void asg::AsgService::print ( ) const

virtualinherited

Print the state of the service.

Implements asg::IAsgService.

Definition at line 48 of file AsgService.cxx.

  {
    ATH_MSG_INFO( "AsgService " << name() << " @ " << this );
    return;
  }

queryInterface()

StatusCode asg::AsgService::queryInterface ( const InterfaceID &  riid, void **  ppvi  )

virtualinherited

query interface for gaudi

Definition at line 57 of file AsgService.cxx.

  {
    for (const auto& interface : m_interfaces)
    {
      if (riid == interface.first())
      {
        *ppvi = interface.second (this);
        addRef();
        return StatusCode::SUCCESS;
      }
    }
    return AsgServiceBase::queryInterface (riid, ppvi);
  }

Member Data Documentation

m_env

std::unique_ptr< Ort::Env > AthOnnx::OnnxRuntimeSvc::m_env

private

Global runtime environment for Onnx Runtime.

Definition at line 52 of file OnnxRuntimeSvc.h.

m_interfaces

std::vector<std::pair<const InterfaceID& (*)(),void *(*)(AsgService*)> > asg::AsgService::m_interfaces

privateinherited

list of interfaces we have

Definition at line 68 of file AsgService.h.

---

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

• OnnxRuntimeSvc.h
• OnnxRuntimeSvc.cxx