Loading Documentation/Examples/Matrices/LambdaMatrix/LambdaMatrixExample_getRow.cu 0 → 100644 +75 −0 Original line number Diff line number Diff line #include <iostream> #include <TNL/Matrices/LambdaMatrix.h> #include <TNL/Devices/Host.h> #include <TNL/Devices/Cuda.h> template< typename Device > void laplaceOperatorMatrix() { /*** * Set matrix representing approximation of the Laplace operator on regular * grid using the finite difference method. */ const int gridSize( 4 ); const int matrixSize = gridSize * gridSize; auto rowLengths = [=] __cuda_callable__ ( const int rows, const int columns, const int rowIdx ) -> int { const int gridRow = rowIdx / gridSize; // coordinates in the numerical grid const int gridColumn = rowIdx % gridSize; if( gridRow == 0 || gridRow == gridSize - 1 || // boundary grid node gridColumn == 0 || gridColumn == gridSize - 1 ) return 1; return 5; }; auto matrixElements = [=] __cuda_callable__ ( const int rows, const int columns, const int rowIdx, const int localIdx, int& columnIdx, double& value) { const int gridRow = rowIdx / gridSize; // coordinates in the numerical grid const int gridColumn = rowIdx % gridSize; if( gridRow == 0 || gridRow == gridSize - 1 || // boundary grid node gridColumn == 0 || gridColumn == gridSize - 1 ) { columnIdx = rowIdx; // diagonal element .... value = 1.0; // ... is set to 1 } else // interior grid node { switch( localIdx ) // set diagonal element to 4 { // and the others to -1 case 0: columnIdx = rowIdx - gridSize; value = 1; break; case 1: columnIdx = rowIdx - 1; value = 1; break; case 2: columnIdx = rowIdx; value = -4; break; case 3: columnIdx = rowIdx + 1; value = 1; break; case 4: columnIdx = rowIdx + gridSize; value = 1; break; } } }; auto matrix = TNL::Matrices::LambdaMatrixFactory< double, Device, int >::create( matrixSize, matrixSize, matrixElements, rowLengths ); std::cout << "Laplace operator matrix: " << std::endl << matrix << std::endl; } int main( int argc, char* argv[] ) { std::cout << "Running example on CPU ... " << std::endl; forRowsExample< TNL::Devices::Host >(); #ifdef HAVE_CUDA std::cout << "Running example on CUDA GPU ... " << std::endl; forRowsExample< TNL::Devices::Cuda >(); #endif } Loading
Documentation/Examples/Matrices/LambdaMatrix/LambdaMatrixExample_getRow.cu 0 → 100644 +75 −0 Original line number Diff line number Diff line #include <iostream> #include <TNL/Matrices/LambdaMatrix.h> #include <TNL/Devices/Host.h> #include <TNL/Devices/Cuda.h> template< typename Device > void laplaceOperatorMatrix() { /*** * Set matrix representing approximation of the Laplace operator on regular * grid using the finite difference method. */ const int gridSize( 4 ); const int matrixSize = gridSize * gridSize; auto rowLengths = [=] __cuda_callable__ ( const int rows, const int columns, const int rowIdx ) -> int { const int gridRow = rowIdx / gridSize; // coordinates in the numerical grid const int gridColumn = rowIdx % gridSize; if( gridRow == 0 || gridRow == gridSize - 1 || // boundary grid node gridColumn == 0 || gridColumn == gridSize - 1 ) return 1; return 5; }; auto matrixElements = [=] __cuda_callable__ ( const int rows, const int columns, const int rowIdx, const int localIdx, int& columnIdx, double& value) { const int gridRow = rowIdx / gridSize; // coordinates in the numerical grid const int gridColumn = rowIdx % gridSize; if( gridRow == 0 || gridRow == gridSize - 1 || // boundary grid node gridColumn == 0 || gridColumn == gridSize - 1 ) { columnIdx = rowIdx; // diagonal element .... value = 1.0; // ... is set to 1 } else // interior grid node { switch( localIdx ) // set diagonal element to 4 { // and the others to -1 case 0: columnIdx = rowIdx - gridSize; value = 1; break; case 1: columnIdx = rowIdx - 1; value = 1; break; case 2: columnIdx = rowIdx; value = -4; break; case 3: columnIdx = rowIdx + 1; value = 1; break; case 4: columnIdx = rowIdx + gridSize; value = 1; break; } } }; auto matrix = TNL::Matrices::LambdaMatrixFactory< double, Device, int >::create( matrixSize, matrixSize, matrixElements, rowLengths ); std::cout << "Laplace operator matrix: " << std::endl << matrix << std::endl; } int main( int argc, char* argv[] ) { std::cout << "Running example on CPU ... " << std::endl; forRowsExample< TNL::Devices::Host >(); #ifdef HAVE_CUDA std::cout << "Running example on CUDA GPU ... " << std::endl; forRowsExample< TNL::Devices::Cuda >(); #endif }