testPtrAlgoAsync.dev.cc

Go to the documentation of this file.

#include "HeterogeneousCore/AlpakaInterface/interface/workdivision.h"

#include "testPtrAlgoAsync.h"

namespace ALPAKA_ACCELERATOR_NAMESPACE {
  portabletest::TestProductWithPtr<Device> testPtrAlgoAsync(Queue& queue, int size) {
    portabletest::TestProductWithPtr<Device> ret{size, queue};
    using View = portabletest::TestProductWithPtr<Device>::View;
    alpaka::exec<Acc1D>(
        queue,
        cms::alpakatools::make_workdiv<Acc1D>(1, 1),
        [] ALPAKA_FN_ACC(Acc1D const& acc, View view) {
          if (cms::alpakatools::once_per_grid(acc)) {
            portabletest::setPtrInTestProductWithPtr(view);
          }
        },
        ret.view());
    alpaka::exec<Acc1D>(
        queue,
        cms::alpakatools::make_workdiv<Acc1D>(1, size),
        [] ALPAKA_FN_ACC(Acc1D const& acc, View view) {
          for (auto i : cms::alpakatools::uniform_elements(acc)) {
            view.ptr()[i] = 2 * i + 1;
          }
        },
        ret.view());
    return ret;
  }
}  // namespace ALPAKA_ACCELERATOR_NAMESPACE