Skip to content
Snippets Groups Projects
Commit 6514e481 authored by Jakub Klinkovský's avatar Jakub Klinkovský
Browse files

Simplified lambda function and local variables in ArrayView::forElements

parent b34bb8a5
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
src/TNL/Containers/ArrayView.hpp
+ 6
− 14
View file @ 6514e481
...@@ -347,15 +347,11 @@ forElements( IndexType begin, IndexType end, Function&& f ) ...@@ -347,15 +347,11 @@ forElements( IndexType begin, IndexType end, Function&& f )
if( ! this->data ) if( ! this->data )
return; return;
ValueType* data = this->getData(); auto g = [] __cuda_callable__ ( IndexType i, Function f, ArrayView view )
const IndexType size = this->getSize();
auto g = [=] __cuda_callable__ ( IndexType i ) mutable
{ {
TNL_ASSERT_GE( i, 0, "Element index must be non-negative." ); f( i, view[ i ] );
TNL_ASSERT_LT( i, size, "Element index is out of bounds." );
f( i, data[ i ] );
}; };
Algorithms::ParallelFor< DeviceType >::exec( begin, end, g ); Algorithms::ParallelFor< DeviceType >::exec( begin, end, g, f, *this );
} }
template< typename Value, template< typename Value,
...@@ -369,15 +365,11 @@ forElements( IndexType begin, IndexType end, Function&& f ) const ...@@ -369,15 +365,11 @@ forElements( IndexType begin, IndexType end, Function&& f ) const
if( ! this->data ) if( ! this->data )
return; return;
const ValueType* data = this->getData(); auto g = [] __cuda_callable__ ( IndexType i, Function f, ArrayView view )
const IndexType size = this->getSize();
auto g = [=] __cuda_callable__ ( IndexType i )
{ {
TNL_ASSERT_GE( i, 0, "Element index must be non-negative." ); f( i, view[ i ] );
TNL_ASSERT_LT( i, size, "Element index is out of bounds." );
f( i, data[ i ] );
}; };
Algorithms::ParallelFor< DeviceType >::exec( begin, end, g ); Algorithms::ParallelFor< DeviceType >::exec( begin, end, g, f, *this );
} }
template< typename Value, template< typename Value,
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment