Skip to content

Commit ea2e5ac

Browse files
committed
Clean up LAMBDA
1 parent 0b55a32 commit ea2e5ac

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

cajita/src/Cajita_SparseMapDynamicPartitioner.hpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -60,15 +60,16 @@ class SparseMapDynamicPartitionerWorkloadMeasurer
6060
//! \brief Called by DynamicPartitioner to compute workload
6161
void compute( Kokkos::View<int***, memory_space>& workload ) override
6262
{
63+
const SparseMapType& sparseMap_proxy = sparseMap;
6364
Kokkos::parallel_for(
6465
"compute_local_workload_sparsmap",
6566
Kokkos::RangePolicy<execution_space>( 0, sparseMap.capacity() ),
6667
KOKKOS_LAMBDA( uint32_t i ) {
67-
if ( sparseMap.valid_at( i ) )
68+
if ( sparseMap_proxy.valid_at( i ) )
6869
{
69-
auto key = sparseMap.key_at( i );
70+
auto key = sparseMap_proxy.key_at( i );
7071
int ti, tj, tk;
71-
sparseMap.key2ijk( key, ti, tj, tk );
72+
sparseMap_proxy.key2ijk( key, ti, tj, tk );
7273
Kokkos::atomic_increment(
7374
&workload( ti + 1, tj + 1, tk + 1 ) );
7475
}

0 commit comments

Comments
 (0)