ExampleMLInferenceWithTriton.h

Go to the documentation of this file.

// Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
 
#ifndef ATHEXTRITON_EXAMPLEMLINFERENCEWITHTRITON_H
#define ATHEXTRITON_EXAMPLEMLINFERENCEWITHTRITON_H
 
// Framework include(s).
#include "AthOnnxInterfaces/IAthInferenceTool.h"
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
 
// System include(s).
#include <string>
#include <vector>
 
namespace AthInfer {

class ExampleMLInferenceWithTriton : public AthReentrantAlgorithm {
 
 public:
  using AthReentrantAlgorithm::AthReentrantAlgorithm;


  virtual StatusCode initialize() override;
  virtual StatusCode execute(const EventContext& ctx) const override;

 
 private:

  Gaudi::Property<std::string> m_pixelFileName{
      this, "InputDataPixel", "dev/MLTest/2020-03-31/t10k-images-idx3-ubyte",
      "Name of the input pixel file to load"};

  Gaudi::Property<int> m_batchSize{this, "BatchSize", 1,
                                   "No. of elements/example in a batch"};

  ToolHandle<AthInfer::IAthInferenceTool> m_tritonTool{
      this, "InferenceTool", "AthInfer::TritonTool", "Triton client tool"};
 
  std::vector<std::vector<std::vector<float>>> m_input_tensor_values_notFlat;
};
 
}  // namespace AthInfer
#endif  // ATHEXTRITON_EXAMPLEMLINFERENCEWITHTRITON_H