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

DataFormats/PortableTestObjects/interface/MaskSoA.h

20 строк611 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_MaskSoA_h
#define DataFormats_PortableTestObjects_interface_MaskSoA_h

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

#include "DataFormats/SoATemplate/interface/SoALayout.h"

namespace portabletest {

  using PartMask = Eigen::Vector<uint8_t, 3>;
  GENERATE_SOA_LAYOUT(MaskLayout, SOA_EIGEN_COLUMN(PartMask, mask));
  GENERATE_SOA_LAYOUT(ScalarMaskLayout, SOA_SCALAR(uint8_t, scalar_mask))

  using MaskSoA = MaskLayout<>;
  using ScalarMaskSoA = ScalarMaskLayout<>;

}  // namespace portabletest

#endif  // DataFormats_PortableTestObjects_interface_MaskSoA_h