hpc_config.h

// -*-c++-*-
#ifndef STORMM_HPC_CONFIG_H
#define STORMM_HPC_CONFIG_H
 
#ifdef STORMM_USE_CUDA
#include <cusolverDn.h>
#include "cublas_v2.h"
#endif
#include "copyright.h"
#include "Constants/behavior.h"
#include "gpu_details.h"
 
namespace stormm {
namespace card {
 
using constants::ExceptionResponse;

class HpcConfig {
public:
  
  HpcConfig(ExceptionResponse policy = ExceptionResponse::DIE);

  ~HpcConfig();

  int getOverallGpuCount() const;

  int getAvailableGpuCount() const;

  int getSupportedGpuCount() const;

  GpuDetails getGpuInfo(int gpu_index) const;

  std::vector<int> getGpuDevice(int requested_count) const;
 
#ifdef STORMM_USE_HPC
  cublasHandle_t getCuBlasHandle() const;

  cusolverDnHandle_t getCuSolverHandle() const;
#endif
  
private:
  int overall_gpu_count;              
  int available_gpu_count;            
  int supported_gpu_count;            
  std::vector<GpuDetails> gpu_list;   
#ifdef STORMM_USE_HPC
  cublasHandle_t cublas_handle;       
  cusolverDnHandle_t cusolver_handle; 
#endif
};

const std::vector<GpuDetails> queryGpuStats(ExceptionResponse policy = ExceptionResponse::DIE);
 
} // namespace card
} // namespace stormm
 
#endif