DataFormats/PortableTestObjects/interface/LogitsSoA.h - master - githubmirror/cmssw

DataFormats/PortableTestObjects/interface/LogitsSoA.h

19 строк616 B

Christine Zeh

Integration of PyTorch based portable (direct) ML inference with Alpaka-based heterogeneous core

21 ноя 2025, 14:01

21 ноя 2025, 14:01a7218f1

#ifndef DataFormats_PortableTestObjects_interface_LogitsSoA_h
#define DataFormats_PortableTestObjects_interface_LogitsSoA_h

#include <Eigen/Core>
#include <Eigen/Dense>

#include "DataFormats/Common/interface/StdArray.h"
#include "DataFormats/SoATemplate/interface/SoACommon.h"
#include "DataFormats/SoATemplate/interface/SoALayout.h"

namespace portabletest {

  using LogitsType = Eigen::Vector<float, 10>;
  GENERATE_SOA_LAYOUT(LogitsLayout, SOA_EIGEN_COLUMN(LogitsType, logits))
  using LogitsSoA = LogitsLayout<>;

}  // namespace portabletest

#endif  // DataFormats_PortableTestObjects_interface_LogitsSoA_h