ALPAKA_ACCELERATOR_NAMESPACE::TestAlgoKernelUpdateMulti3 Class Reference

Public Member Functions
template<typename TAcc , typename = std::enable_if_t<alpaka::isAccelerator<TAcc>>>
ALPAKA_FN_ACC void	operator() (TAcc const &acc, portabletest::TestSoA::ConstView input, portabletest::TestSoA2::ConstView input2, portabletest::TestSoA3::ConstView input3, AlpakaESTestDataEDevice::ConstView esData, portabletest::TestSoA::View output, portabletest::TestSoA2::View output2, portabletest::TestSoA3::View output3) const

Detailed Description

Definition at line 220 of file TestAlgo.dev.cc.

Member Function Documentation

operator()()

template<typename TAcc , typename = std::enable_if_t<alpaka::isAccelerator<TAcc>>>

ALPAKA_FN_ACC void ALPAKA_ACCELERATOR_NAMESPACE::TestAlgoKernelUpdateMulti3::operator() ( TAcc const &  acc, portabletest::TestSoA::ConstView  input, portabletest::TestSoA2::ConstView  input2, portabletest::TestSoA3::ConstView  input3, AlpakaESTestDataEDevice::ConstView  esData, portabletest::TestSoA::View  output, portabletest::TestSoA2::View  output2, portabletest::TestSoA3::View  output3  ) const

inline

Definition at line 223 of file TestAlgo.dev.cc.

References cms::alpakatools::elements_with_stride(), mps_fire::i, input, input2, cms::alpakatools::once_per_grid(), skim900GeV_StreamA_MinBiasPD_cfg::output2, skim900GeV_StreamA_MinBiasPD_cfg::output3, cms::alpakatest::AlpakaESTestDataE< TDev >::ConstView::size(), cms::alpakatest::AlpakaESTestDataE< TDev >::ConstView::val(), cms::alpakatest::AlpakaESTestDataE< TDev >::ConstView::val2(), and x.

                                                                          {
      // set this only once in the whole kernel grid
      if (once_per_grid(acc)) {
        output.r() = input.r();
        output2.r2() = input2.r2();
        output3.r3() = input3.r3();
      }

      // make a strided loop over the kernel grid, covering up to "size" elements
      for (int32_t i : elements_with_stride(acc, output.metadata().size())) {
        double x = input[i].x();
        if (i < esData.size()) {
          x += esData.val(i) + esData.val2(i);
          if (0 == i)
            printf("Setting x[0] to %f\n", x);
        }
        output[i] = {x, input[i].y(), input[i].z(), input[i].id(), input[i].flags(), input[i].m()};
      }
      for (int32_t i : elements_with_stride(acc, output2.metadata().size())) {
        double x2 = input2[i].x2();
        if (i < esData.size()) {
          x2 += esData.val(i) + esData.val2(i);
        }
        output2[i] = {x2, input2[i].y2(), input2[i].z2(), input2[i].id2(), input2[i].m2()};
      }
      for (int32_t i : elements_with_stride(acc, output3.metadata().size())) {
        double x3 = input3[i].x3();
        if (i < esData.size()) {
          x3 += esData.val(i) + esData.val2(i);
        }
        output3[i] = {x3, input3[i].y3(), input3[i].z3(), input3[i].id3(), input3[i].m3()};
      }
    }