HeterogeneousCore/TrivialSerialisation/interface/alpaka/SerialiserFactoryDevice.h - master - githubmirror/cmssw

HeterogeneousCore/TrivialSerialisation/interface/alpaka/SerialiserFactoryDevice.h

30 строк2 KB

Mario Gonzalez

Extend the HeterogeneousCore/TrivialSerialisation mechanism to device products

31 мар 2026, 15:33

31 мар 2026, 15:33be97c80

#ifndef HeterogeneousCore_TrivialSerialisation_interface_alpaka_SerialiserFactoryDevice_h
#define HeterogeneousCore_TrivialSerialisation_interface_alpaka_SerialiserFactoryDevice_h

#include "DataFormats/Common/interface/DeviceProduct.h"
#include "FWCore/PluginManager/interface/PluginFactory.h"
#include "FWCore/Utilities/interface/stringize.h"
#include "HeterogeneousCore/AlpakaInterface/interface/config.h"
#include "HeterogeneousCore/TrivialSerialisation/interface/alpaka/Serialiser.h"
#include "HeterogeneousCore/TrivialSerialisation/interface/alpaka/SerialiserBase.h"

namespace ALPAKA_ACCELERATOR_NAMESPACE::ngt {
  using SerialiserFactoryDevice = edmplugin::PluginFactory<SerialiserBase*()>;
}  // namespace ALPAKA_ACCELERATOR_NAMESPACE::ngt

// Helper macro to define Serialiser plugins.
//
// TYPE is the inner product type (e.g. PortableDeviceCollection<...>), not
// wrapped in DeviceProduct, and without ALPAKA_ACCELERATOR_NAMESPACE:: attached
// to it (it is attached here). The plugin is registered under both the mangled
// typeid name and EDM_STRINGIZE(TYPE). EDM_STRINGIZE(TYPE) is more
// human-readable, and thus more suitable for Python configuration files.
#define DEFINE_TRIVIAL_SERIALISER_PLUGIN_DEVICE(TYPE)                                                   \
  DEFINE_EDM_PLUGIN(ALPAKA_ACCELERATOR_NAMESPACE::ngt::SerialiserFactoryDevice,                         \
                    ALPAKA_ACCELERATOR_NAMESPACE::ngt::Serialiser<ALPAKA_ACCELERATOR_NAMESPACE::TYPE>,  \
                    typeid(edm::DeviceProduct<ALPAKA_ACCELERATOR_NAMESPACE::TYPE>).name());             \
  DEFINE_EDM_PLUGIN2(ALPAKA_ACCELERATOR_NAMESPACE::ngt::SerialiserFactoryDevice,                        \
                     ALPAKA_ACCELERATOR_NAMESPACE::ngt::Serialiser<ALPAKA_ACCELERATOR_NAMESPACE::TYPE>, \
                     EDM_STRINGIZE(TYPE))

#endif  // HeterogeneousCore_TrivialSerialisation_interface_alpaka_SerialiserFactoryDevice_h