DataFormats/PortableTestObjects/interface/alpaka/MaskDeviceCollection.h - master - githubmirror/cmssw

DataFormats/PortableTestObjects/interface/alpaka/MaskDeviceCollection.h

21 строка1 KB

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_alpaka_MaskDeviceCollection_h
#define DataFormats_PortableTestObjects_interface_alpaka_MaskDeviceCollection_h

#include "DataFormats/Portable/interface/alpaka/PortableCollection.h"
#include "HeterogeneousCore/AlpakaInterface/interface/config.h"
#include "DataFormats/PortableTestObjects/interface/MaskHostCollection.h"
#include "DataFormats/PortableTestObjects/interface/MaskSoA.h"

namespace ALPAKA_ACCELERATOR_NAMESPACE::portabletest {

  using namespace ::portabletest;

  using MaskDeviceCollection = PortableCollection<MaskSoA>;
  using ScalarMaskDeviceCollection = PortableCollection<ScalarMaskSoA>;

}  // namespace ALPAKA_ACCELERATOR_NAMESPACE::portabletest

ASSERT_DEVICE_MATCHES_HOST_COLLECTION(portabletest::MaskDeviceCollection, portabletest::MaskHostCollection);
ASSERT_DEVICE_MATCHES_HOST_COLLECTION(portabletest::ScalarMaskDeviceCollection, portabletest::ScalarMaskHostCollection);

#endif  // DataFormats_PortableTestObjects_interface_alpaka_MaskDeviceCollection_h