plugins/alpaka/AlpakaTestKernelAdditionAlgo.dev.cc

0001 #include <cstdint>
0002
0003 #include <alpaka/alpaka.hpp>
0004
0005 #include "HeterogeneousCore/AlpakaInterface/interface/config.h"
0006 #include "HeterogeneousCore/AlpakaInterface/interface/workdivision.h"
0007 #include "HeterogeneousTest/AlpakaKernel/interface/alpaka/DeviceAdditionKernel.h"
0008
0009 #include "AlpakaTestKernelAdditionAlgo.h"
0010
0011 namespace ALPAKA_ACCELERATOR_NAMESPACE::HeterogeneousTestAlpakaKernelPlugins {
0012
0013   void wrapper_add_vectors_f(Queue& queue,
0014                              const float* __restrict__ in1,
0015                              const float* __restrict__ in2,
0016                              float* __restrict__ out,
0017                              uint32_t size) {
0018     alpaka::exec<Acc1D>(
0019         queue, cms::alpakatools::make_workdiv<Acc1D>(32, 32), test::KernelAddVectorsF{}, in1, in2, out, size);
0020   }
0021
0022 }  // namespace ALPAKA_ACCELERATOR_NAMESPACE::HeterogeneousTestAlpakaKernelPlugins