CudaMgr.cpp File Reference

#include "CudaMgr/CudaMgr.h"
#include "QueryEngine/NvidiaKernel.h"
#include "Shared/boost_stacktrace.hpp"
#include <algorithm>
#include <cassert>
#include <iostream>
#include <stdexcept>
#include "Logger/Logger.h"

Include dependency graph for CudaMgr.cpp:

Go to the source code of this file.

Namespaces
	CudaMgr_Namespace

Functions
std::string	CudaMgr_Namespace::errorMessage (CUresult const status)
std::string	get_cuda_home (void)

Function Documentation

std::string get_cuda_home

(

void

)

Definition at line 404 of file CudaMgr.cpp.

References LOG, and logger::WARNING.

Referenced by UdfCompiler::compileToGpuByteCode().

                              {
  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);
}

Here is the caller graph for this function: