PhysicsTools/PyTorchAlpakaTest/plugins/alpaka/CommonKernels.h - master - githubmirror/cmssw

PhysicsTools/PyTorchAlpakaTest/plugins/alpaka/CommonKernels.h

18 строк959 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 PhysicsTools_PyTorchAlpakaTest_plugins_alpaka_CommonKernels_h
#define PhysicsTools_PyTorchAlpakaTest_plugins_alpaka_CommonKernels_h

#include <alpaka/alpaka.hpp>
#include "DataFormats/PortableTestObjects/interface/alpaka/ParticleDeviceCollection.h"
#include "DataFormats/PortableTestObjects/interface/alpaka/ImageDeviceCollection.h"
#include "DataFormats/PortableTestObjects/interface/alpaka/MaskDeviceCollection.h"
#include "HeterogeneousCore/AlpakaInterface/interface/config.h"

namespace ALPAKA_ACCELERATOR_NAMESPACE::torchtest::kernels {

  void randomFillParticleCollection(Queue& queue, portabletest::ParticleDeviceCollection& particles);
  void randomFillImageCollection(Queue& queue, portabletest::ImageDeviceCollection& images);
  void fillMask(Queue& queue, portabletest::MaskDeviceCollection& mask);

}  // namespace ALPAKA_ACCELERATOR_NAMESPACE::torchtest::kernels

#endif  // PhysicsTools_PyTorchAlpakaTest_plugins_alpaka_CommonKernels_h