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 60 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 63 of file TestAlgo.dev.cc.

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

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