Compadre_KokkosParser.hpp

Go to the documentation of this file.

#ifndef _COMPADRE_KOKKOSPARSER_HPP_
#define _COMPADRE_KOKKOSPARSER_HPP_

#include "Compadre_Config.h"
#include "Compadre_Typedefs.hpp"

namespace Compadre {

/*! \class KokkosParser
    \brief Class handling Kokkos command line arguments and returning parameters.
*/
class KokkosParser {

private:

  int _num_threads;
  int _numa;
  int _device;
  int _ngpu;

  bool _called_initialize;

  // prevent default constructor
  KokkosParser();

public:

  // call with command line arguments
  KokkosParser(int argc, char* args[], bool print_status = false);

  // call with integer arguments
  KokkosParser(int num_threads = -1, int numa = -1, int device = -1, int ngpu = -1, bool print_status = false);

  // destructor
  ~KokkosParser() {
      // clean-up Kokkos
      if (_called_initialize) {
          this->finalize();
      } 
  };

  // initialize Kokkos if not already initialized using
  // arguments provided at object construction
  int initialize();
  
  // finalize Kokkos if this object initialized it
  // or if hard_finalize is true
  int finalize(bool hard_finalize = false);

  // retrieve arguments from a previous initialized Kokkos state
  void retrievePreviouslyInstantiatedKokkosInitArguments();

  // use this object's state to create a Kokkos object used
  // later for initialization
  Kokkos::InitArguments createInitArguments() const;

  int getNumberOfThreads() const { return _num_threads; }
  int getNuma() const { return _numa; }
  int getDeviceID() const { return _device; }
  int getNumberOfGPUs() const { return _ngpu; }

  // prints status
  void status() const;

  // prohibit using the assignment constructor
  KokkosParser& operator=( const KokkosParser& ) = delete;
  //KokkosParser( const KokkosParser& ) = delete;

};

} // Compadre

#endif