CMS 3D CMS Logo

List of all members | Public Member Functions
ALPAKA_ACCELERATOR_NAMESPACE::TestAlgoMultiKernel3 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::TestDeviceMultiCollection3::View< 2 > view, double xvalue) const
 

Detailed Description

Definition at line 61 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::TestAlgoMultiKernel3::operator() ( TAcc const &  acc,
portabletest::TestDeviceMultiCollection3::View< 2 >  view,
double  xvalue 
) const
inline

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

References cms::alpakatools::elements_with_stride(), mps_fire::i, and makeMuonMisalignmentScenario::matrix.

66  {
67  // global index of the thread within the grid
68  const int32_t thread = alpaka::getIdx<alpaka::Grid, alpaka::Threads>(acc)[0u];
69  const portabletest::Matrix matrix{{1, 2, 3, 4, 5, 6}, {2, 4, 6, 8, 10, 12}, {3, 6, 9, 12, 15, 18}};
70 
71  // set this only once in the whole kernel grid
72  if (thread == 0) {
73  view.r3() = 3.;
74  }
75 
76  // make a strided loop over the kernel grid, covering up to "size" elements
77  for (int32_t i : elements_with_stride(acc, view.metadata().size())) {
78  view[i] = {xvalue, 0., 0., i, matrix * i};
79  }
80  }
Eigen::Matrix< double, 3, 6 > Matrix
Definition: TestSoA.h:19
ALPAKA_FN_ACC auto elements_with_stride(TAcc const &acc, TArgs... args)
Definition: workdivision.h:346