From 25d43763f10fe3f6f19851daa152f035e2a08439 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tom=C3=A1=C5=A1=20Oberhuber?= <oberhuber.tomas@gmail.com>
Date: Fri, 3 Jan 2020 11:07:16 +0100
Subject: [PATCH] Reimplementation of
Matrices::Dense::getNumberOfNonzeroMatrixElements.
---
src/TNL/Matrices/Dense.hpp | 11 +++++------
1 file changed, 5 insertions(+), 6 deletions(-)
diff --git a/src/TNL/Matrices/Dense.hpp b/src/TNL/Matrices/Dense.hpp
index 70e5018dde..7e6dec9ce6 100644
--- a/src/TNL/Matrices/Dense.hpp
+++ b/src/TNL/Matrices/Dense.hpp
@@ -130,12 +130,11 @@ template< typename Real,
typename RealAllocator >
Index Dense< Real, Device, Index, RowMajorOrder, RealAllocator >::getNumberOfNonzeroMatrixElements() const
{
- IndexType nonzeroElements( 0 );
- for( IndexType row = 0; row < this->getRows(); row++ )
- for( IndexType column = 0; column < this->getColumns(); column++ )
- if( this->getElement( row, column ) != 0 )
- nonzeroElements++;
- return nonzeroElements;
+ const auto values_view = this->values.getConstView();
+ auto fetch = [=] __cuda_callable__ ( const IndexType i ) -> IndexType {
+ return ( values_view[ i ] != 0.0 );
+ };
+ return Algorithms::Reduction< DeviceType >::reduce( this->values.getSize(), std::plus<>{}, fetch, 0 );
}
template< typename Real,
--
GitLab