CudaMgr.h File Reference

#include <cstdlib>
#include <mutex>
#include <string>
#include <vector>
#include "Logger/Logger.h"
#include "Shared/DeviceGroup.h"
#include "Shared/nocuda.h"

Include dependency graph for CudaMgr.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	CudaMgr_Namespace::DeviceProperties
class	CudaMgr_Namespace::CudaMgr

Namespaces
	CudaMgr_Namespace

Enumerations
enum	CudaMgr_Namespace::NvidiaDeviceArch {   CudaMgr_Namespace::NvidiaDeviceArch::Kepler, CudaMgr_Namespace::NvidiaDeviceArch::Maxwell, CudaMgr_Namespace::NvidiaDeviceArch::Pascal, CudaMgr_Namespace::NvidiaDeviceArch::Volta,   CudaMgr_Namespace::NvidiaDeviceArch::Turing, CudaMgr_Namespace::NvidiaDeviceArch::Ampere }

Functions
std::string	get_cuda_home (void)
std::string	get_cuda_libdevice_dir (void)

Function Documentation

std::string get_cuda_home

(

void

)

Definition at line 465 of file CudaMgr.cpp.

References LOG, and logger::WARNING.

                              {
  static const char* CUDA_DEFAULT_PATH = "/usr/local/cuda";
  const char* env = nullptr;

  if (!(env = getenv("CUDA_HOME")) && !(env = getenv("CUDA_DIR"))) {
    // check if the default CUDA directory exists: /usr/local/cuda
    if (boost::filesystem::exists(boost::filesystem::path(CUDA_DEFAULT_PATH))) {
      env = CUDA_DEFAULT_PATH;
    }
  }

  if (env == nullptr) {
    LOG(WARNING) << "Could not find CUDA installation path: environment variables "
                    "CUDA_HOME or CUDA_DIR are not defined";
    return "";
  }

  // check if the CUDA directory is sensible:
  auto cuda_include_dir = env + std::string("/include");
  auto cuda_h_file = cuda_include_dir + "/cuda.h";
  if (!boost::filesystem::exists(boost::filesystem::path(cuda_h_file))) {
    LOG(WARNING) << "cuda.h does not exist in `" << cuda_include_dir << "`. Discarding `"
                 << env << "` as CUDA installation path.";
    return "";
  }

  return std::string(env);
}

std::string get_cuda_libdevice_dir

(

void

)

Definition at line 494 of file CudaMgr.cpp.

References LOG, and logger::WARNING.

Referenced by Executor::initialize_extension_module_sources().

                                       {
  static const char* CUDA_DEFAULT_PATH = "/usr/local/cuda";
  const char* env = nullptr;

  if (!(env = getenv("CUDA_HOME")) && !(env = getenv("CUDA_DIR"))) {
    // check if the default CUDA directory exists: /usr/local/cuda
    if (boost::filesystem::exists(boost::filesystem::path(CUDA_DEFAULT_PATH))) {
      env = CUDA_DEFAULT_PATH;
    }
  }

  if (env == nullptr) {
    LOG(WARNING) << "Could not find CUDA installation path: environment variables "
                    "CUDA_HOME or CUDA_DIR are not defined";
    return "";
  }

  // check if the CUDA directory is sensible:
  auto libdevice_dir = env + std::string("/nvvm/libdevice");
  auto libdevice_bc_file = libdevice_dir + "/libdevice.10.bc";
  if (!boost::filesystem::exists(boost::filesystem::path(libdevice_bc_file))) {
    LOG(WARNING) << "`" << libdevice_bc_file << "` does not exist. Discarding `" << env
                 << "` as CUDA installation path with libdevice.";
    return "";
  }

  return libdevice_dir;
}

Here is the caller graph for this function: