cuda/init.hpp

Go to the documentation of this file.

/*
 * Copyright (C) 2010-2022 The ESPResSo project
 *
 * This file is part of ESPResSo.
 *
 * ESPResSo is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * ESPResSo is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
#pragma once
 
#include "config/config.hpp"
 
#ifdef ESPRESSO_CUDA
 
#include "utils.hpp"
 
#include <boost/serialization/access.hpp>
 
#include <cstddef>
#include <string>
#include <utility>
#include <vector>
 
/** Struct to hold information relevant to ESPResSo
 *  about GPUs. Should contain only fixed-length plain
 *  old datatypes, as it is intended for MPI communication.
 */
struct EspressoGpuDevice {
  /** Local CUDA device id */
  int id;
  /** Local CUDA device name */
  std::string name;
  /** Node identification */
  std::string proc_name;
  /** MPI process identification */
  int node;
  /** Compute capability (major) */
  int compute_capability_major;
  /** Compute capability (minor) */
  int compute_capability_minor;
  /** Total Memory */
  std::size_t total_memory;
  /** Number of cores */
  int n_cores;
 
private:
  friend boost::serialization::access;
  template <typename Archive>
  void serialize(Archive &ar, unsigned int const /* version */) {
    ar & id & name & proc_name & node & compute_capability_major &
        compute_capability_minor & total_memory & n_cores;
  }
};
 
/** Initializes the CUDA stream.
 */
void cuda_init();
 
/** Get the number of CUDA devices on the local host.
 *
 *  @return the number of GPUs.
 */
int cuda_get_n_gpus();
 
/** Check that a given GPU has compute capability.
 *  The minimal compute capability required by ESPResSo is
 *  \ref computeCapabilityMinMajor . \ref computeCapabilityMinMinor .
 *
 *  @param dev CUDA device number
 *  @return @c false if the GPU meets the requirements, else @c true.
 */
bool cuda_check_gpu_compute_capability(int dev);
 
/** Get the name of a CUDA device.
 *
 *  @param[in]  dev the CUDA device number to ask the name for
 */
std::string cuda_get_gpu_name(int dev);
 
/** Choose a device for future CUDA computations.
 *
 *  @param dev the device to use
 */
void cuda_set_device(int dev);
 
/** Get the current CUDA device.
 *
 *  @return the current device's number.
 */
int cuda_get_device();
 
/** Test if communication to the CUDA device works.
 *  @return @c false on success, else @c true.
 */
bool cuda_test_device_access();
 
/**
 * Check that a device is available, that its compute capability
 * is sufficient for ESPResSo, and that data can be written to
 * and read from it. Otherwise, throw an exception.
 */
void cuda_check_device();
 
/** Gather unique list of CUDA devices on all nodes.
 *  @return vector of device properties.
 */
std::vector<EspressoGpuDevice> cuda_gather_gpus();
 
/** Get properties of a CUDA device
 *  @param dev CUDA device number
 */
EspressoGpuDevice cuda_get_device_props(int dev);
 
/** @brief Called on program start. */
void cuda_on_program_start();
 
namespace detail {
std::pair<int, std::string> get_node_info();
} // namespace detail
 
#endif // ESPRESSO_CUDA