Commit 93fd2bec authored by Jakub Klinkovský's avatar Jakub Klinkovský
Browse files

Benchmarks: allow to run linear-solvers benchmarks only on GPU

parent c4d018b8

src/Benchmarks/LinearSolvers/benchmarks.h

+21 −0
Original line number Diff line number Diff line
@@ -39,12 +39,29 @@ void barrier( const DistributedContainers::DistributedMatrix< Matrix, Communicat 
   Communicator::Barrier( matrix.getCommunicationGroup() );

}



template< typename Device >

bool checkDevice( const Config::ParameterContainer& parameters )

{

   const String device = parameters.getParameter< String >( "devices" );

   if( device == "all" )

      return true;

   if( std::is_same< Device, Devices::Host >::value && device == "host" )

      return true;

   if( std::is_same< Device, Devices::Cuda >::value && device == "cuda" )

      return true;

   return false;

}



template< template<typename> class Preconditioner, typename Matrix >

void

benchmarkPreconditionerUpdate( Benchmark& benchmark,

                               const Config::ParameterContainer& parameters,

                               const SharedPointer< Matrix >& matrix )

{

   // skip benchmarks on devices which the user did not select

   if( ! checkDevice< typename Matrix::DeviceType >( parameters ) )

      return;



   barrier( matrix );

   const char* performer = getPerformer< typename Matrix::DeviceType >();

   Preconditioner< Matrix > preconditioner;
@@ -67,6 +84,10 @@ benchmarkSolver( Benchmark& benchmark, 
                 const Vector& x0,

                 const Vector& b )

{

   // skip benchmarks on devices which the user did not select

   if( ! checkDevice< typename Matrix::DeviceType >( parameters ) )

      return;



   barrier( matrix );

   const char* performer = getPerformer< typename Matrix::DeviceType >();

src/Benchmarks/LinearSolvers/tnl-benchmark-linear-solvers.h

+6 −0
Original line number Diff line number Diff line
@@ -497,6 +497,12 @@ configSetup( Config::ConfigDescription& config ) 
   config.addEntry< String >( "solvers", "Comma-separated list of solvers to run benchmarks for. Options: gmres, cwygmres, tfqmr, bicgstab, bicgstab-ell.", "all" );

   config.addEntry< String >( "preconditioners", "Comma-separated list of preconditioners to run benchmarks for. Options: jacobi, ilu0, ilut.", "all" );

   config.addEntry< bool >( "with-preconditioner-update", "Run benchmark for the preconditioner update.", true );

   config.addEntry< String >( "devices", "Run benchmarks on these devices.", "all" );

   config.addEntryEnum( "all" );

   config.addEntryEnum( "host" );

   #ifdef HAVE_CUDA

   config.addEntryEnum( "cuda" );

   #endif



   config.addDelimiter( "Device settings:" );

   Devices::Host::configSetup( config );