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

DataFormats/PortableTestObjects/interface/alpaka/LogitsDeviceCollection.h

26 строк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_LogitsDeviceCollection_h
#define DataFormats_PortableTestObjects_interface_alpaka_LogitsDeviceCollection_h

#include "DataFormats/Portable/interface/alpaka/PortableCollection.h"
#include "DataFormats/PortableTestObjects/interface/LogitsHostCollection.h"
#include "DataFormats/PortableTestObjects/interface/LogitsSoA.h"
#include "HeterogeneousCore/AlpakaInterface/interface/config.h"

namespace ALPAKA_ACCELERATOR_NAMESPACE {

  namespace portabletest {

    // make the names from the top-level portabletest namespace visible for unqualified lookup
    // inside the ALPAKA_ACCELERATOR_NAMESPACE::portabletest namespace
    using namespace ::portabletest;

    using LogitsDeviceCollection = PortableCollection<LogitsSoA>;

  }  // namespace portabletest

}  // namespace ALPAKA_ACCELERATOR_NAMESPACE

// heterogeneous ml data checks
ASSERT_DEVICE_MATCHES_HOST_COLLECTION(portabletest::LogitsDeviceCollection, portabletest::LogitsHostCollection);

#endif  // DataFormats_PortableTestObjects_interface_alpaka_LogitsDeviceCollection_h