OnnxRuntimeSvc.h

Go to the documentation of this file.

// Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
#ifndef ATHONNX_ONNXRUNTIMESVC_H
#define ATHONNX_ONNXRUNTIMESVC_H
 
// Local include(s).
#include "AthOnnxInterfaces/IOnnxRuntimeSvc.h"
 
// Framework include(s).
#include <AsgServices/AsgService.h>
#include "AsgTools/PropertyWrapper.h"
 
// ONNX include(s).
#include <onnxruntime_cxx_api.h>
 
// System include(s).
#include <memory>
 
namespace AthOnnx {

   class OnnxRuntimeSvc : public extends<asg::AsgService, IOnnxRuntimeSvc> {
 
   public:

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

      virtual StatusCode initialize() override;
      virtual StatusCode finalize() override;



      virtual Ort::Env& env() const override;

 
   private:
      Gaudi::Property<int> m_logLevel{this, "LogLevel", 2, "ONNX Runtime logging level (0=VERBOSE, 1=INFO, 2=WARNING, 3=ERROR, 4=FATAL)"};
      
      std::unique_ptr< Ort::Env > m_env;
 
   }; // class OnnxRuntimeSvc
 
} // namespace AthOnnx
 
#endif // ATHONNX_ONNXRUNTIMESVC_H