Added examle for MultidiagonalMatrix[,View]::forRows. (55469dd7) · Commits · TNL / tnl-dev

Documentation/Examples/Matrices/MultidiagonalMatrix/CMakeLists.txt

+2 −2

Original line number	Diff line number	Diff line
		@@ -16,7 +16,7 @@ set( COMMON_EXAMPLES
		MultidiagonalMatrixExample_allRowsReduction
		MultidiagonalMatrixExample_forElements
		MultidiagonalMatrixExample_forAllElements
		#MultidiagonalMatrixExample_forRows
		MultidiagonalMatrixExample_forRows
		# MultidiagonalMatrixViewExample_constructor
		MultidiagonalMatrixViewExample_getCompressedRowLengths
		# MultidiagonalMatrixViewExample_getElementsCount
		@@ -29,7 +29,7 @@ set( COMMON_EXAMPLES
		MultidiagonalMatrixViewExample_allRowsReduction
		MultidiagonalMatrixViewExample_forElements
		MultidiagonalMatrixViewExample_forAllElements
		#MultidiagonalMatrixViewExample_forRows
		MultidiagonalMatrixViewExample_forRows
		)

		if( BUILD_CUDA )

Documentation/Examples/Matrices/MultidiagonalMatrix/MultidiagonalMatrixExample_forRows.cpp

0 → 100644

+63 −0

Original line number	Diff line number	Diff line
		#include <iostream>
		#include <TNL/Matrices/MultidiagonalMatrix.h>
		#include <TNL/Devices/Host.h>
		#include <TNL/Devices/Cuda.h>

		template< typename Device >
		void forRowsExample()
		{
		using MatrixType = TNL::Matrices::MultidiagonalMatrix< double, Device >;
		/***
		* Set the following matrix (dots represent zero matrix elements and zeros are
		* padding zeros for memory alignment):
		*
		* 0 / 1 -2 . . . \ -> { 0, 0, 1 }
		* \| -2 1 -2 . . \| -> { 0, 2, 1 }
		* \| . -2 1 -2. . \| -> { 3, 2, 1 }
		* \| . . -2 1 -2 \| -> { 3, 2, 1 }
		* \ . . . -2 1 / -> { 3, 2, 1 }
		*
		* The diagonals offsets are { -1, 0, 1 }.
		*/
		const int size = 5;
		MatrixType matrix(
		size, // number of matrix rows
		size, // number of matrix columns
		{ -2, -1, 0 } ); // matrix diagonals offsets

		auto f = [=] __cuda_callable__ ( typename MatrixType::RowViewType& row ) {
		/***
		* 'forElements' method iterates only over matrix elements lying on given subdiagonals
		* and so we do not need to check anything. The element value can be expressed
		* by the 'localIdx' variable, see the following figure:
		*
		* 0 1 2 <- localIdx values
		* ----------
		* 0 / 1 -2 . . . \ -> { 0, 1, -2 }
		* \| -2 1 -2 . . \| -> { -2, 1, -2 }
		* \| . -2 1 -2. . \| -> { -2, 1, -2 }
		* \| . . -2 1 -2 \| -> { -2, 1, -2 }
		* \ . . . -2 1 / -> { -2, 1, 0 }
		*
		*/
		const int& rowIdx = row.getRowIndex();
		row.setElement( 1, 1.0 );
		if( rowIdx > 0 )
		row.setElement( 0, -2.0 );
		if( rowIdx < size - 1 )
		row.setElement( 2, -2.0 );
		};
		matrix.forAllRows( f );
		std::cout << matrix << std::endl;
		}

		int main( int argc, char* argv[] )
		{
		std::cout << "Creating matrix on host: " << std::endl;
		forRowsExample< TNL::Devices::Host >();

		#ifdef HAVE_CUDA
		std::cout << "Creating matrix on CUDA device: " << std::endl;
		forRowsExample< TNL::Devices::Cuda >();
		#endif
		}

Documentation/Examples/Matrices/MultidiagonalMatrix/MultidiagonalMatrixExample_forRows.cu

0 → 120000

+1 −0

Original line number	Diff line number	Diff line
		MultidiagonalMatrixExample_forRows.cpp
		No newline at end of file

Documentation/Examples/Matrices/MultidiagonalMatrix/MultidiagonalMatrixViewExample_forRows.cpp

0 → 100644

+65 −0

Original line number	Diff line number	Diff line
		#include <iostream>
		#include <TNL/Matrices/MultidiagonalMatrix.h>
		#include <TNL/Devices/Host.h>
		#include <TNL/Devices/Cuda.h>

		template< typename Device >
		void forRowsExample()
		{
		using MatrixType = TNL::Matrices::MultidiagonalMatrix< double, Device >;
		/***
		* Set the following matrix (dots represent zero matrix elements and zeros are
		* padding zeros for memory alignment):
		*
		* 0 / 1 -2 . . . \ -> { 0, 0, 1 }
		* \| -2 1 -2 . . \| -> { 0, 2, 1 }
		* \| . -2 1 -2. . \| -> { 3, 2, 1 }
		* \| . . -2 1 -2 \| -> { 3, 2, 1 }
		* \ . . . -2 1 / -> { 3, 2, 1 }
		*
		* The diagonals offsets are { -1, 0, 1 }.
		*/

		const int size = 5;
		MatrixType matrix(
		size, // number of matrix rows
		size, // number of matrix columns
		{ -2, -1, 0 } ); // matrix diagonals offsets
		auto view = matrix.getView();

		auto f = [=] __cuda_callable__ ( typename MatrixType::RowViewType& row ) {
		/***
		* 'forElements' method iterates only over matrix elements lying on given subdiagonals
		* and so we do not need to check anything. The element value can be expressed
		* by the 'localIdx' variable, see the following figure:
		*
		* 0 1 2 <- localIdx values
		* ----------
		* 0 / 1 -2 . . . \ -> { 0, 1, -2 }
		* \| -2 1 -2 . . \| -> { -2, 1, -2 }
		* \| . -2 1 -2. . \| -> { -2, 1, -2 }
		* \| . . -2 1 -2 \| -> { -2, 1, -2 }
		* \ . . . -2 1 / -> { -2, 1, 0 }
		*
		*/
		const int& rowIdx = row.getRowIndex();
		row.setElement( 1, 1.0 );
		if( rowIdx > 0 )
		row.setElement( 0, -2.0 );
		if( rowIdx < size - 1 )
		row.setElement( 2, -2.0 );
		};
		view.forAllRows( f );
		std::cout << matrix << std::endl;
		}

		int main( int argc, char* argv[] )
		{
		std::cout << "Creating matrix on host: " << std::endl;
		forRowsExample< TNL::Devices::Host >();

		#ifdef HAVE_CUDA
		std::cout << "Creating matrix on CUDA device: " << std::endl;
		forRowsExample< TNL::Devices::Cuda >();
		#endif
		}

Documentation/Examples/Matrices/MultidiagonalMatrix/MultidiagonalMatrixViewExample_forRows.cu

0 → 120000

+1 −0

Original line number	Diff line number	Diff line
		MultidiagonalMatrixViewExample_forRows.cpp
		No newline at end of file

src/TNL/Matrices/details/MultidiagonalMatrixIndexer.h

+2 −2

File changed.

Contains only whitespace changes.