Shuenhoy/vtk-m
1
0
Fork 0
mirror of https://gitlab.kitware.com/vtk/vtk-m synced 2024-09-08 21:33:55 +00:00
Code Issues 2 Actions Packages Projects Releases Wiki Activity

KOKKOS: resolved issue with KOKKOS_LAMBDA

Browse Source
This commit is contained in:
Vicente Adolfo Bolea Sanchez 2021-08-23 18:10:23 -04:00
parent 24621e95ed
commit a6c4e8479f
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
vtkm/cont/kokkos/internal/DeviceAdapterAlgorithmKokkos.h
View File

@ -688,9 +688,17 @@ public:
{ 0, 0, 0 },
{ rangeMax[0], rangeMax[1], rangeMax[2] });
// Calling rangeMax[X] inside KOKKOS_LAMBDA confuses some compilers since
// at first it tries to use the non-const inline vec_base::operator[0]
// method, however, KOKKOS_LAMBDA DOES converts rangeMax to a const
// vec_base. This convertion is somehow catched by the compiler making it
// complain that we are using a non-const method for a const object.
const auto rMax_0 = rangeMax[0];
const auto rMax_1 = rangeMax[1];
Kokkos::parallel_for(
policy, KOKKOS_LAMBDA(vtkm::Id i, vtkm::Id j, vtkm::Id k) {
auto flatIdx = i + (j * rangeMax[0]) + (k * rangeMax[0] * rangeMax[1]);
auto flatIdx = i + (j * rMax_0) + (k * rMax_0 * rMax_1);
functor(vtkm::Id3(i, j, k), flatIdx);
});
CheckForErrors(); // synchronizes