Method get_cuda_include_path

src/core/impl/utils/cuda_helper.cpp:118–160

}  // namespace

std::vector<std::string> mgb::get_cuda_include_path() {
    std::vector<std::string> paths;
    // 1. use CUDA_BIN_PATH
    auto cuda_path = getenv("CUDA_BIN_PATH");
    if (cuda_path) {
        paths.emplace_back(std::string(cuda_path) + PATH_SPLITER + "include");
        paths.emplace_back(
                std::string(cuda_path) + PATH_SPLITER + ".." + PATH_SPLITER +
                "include");
    }

    // 2. use nvcc path
    auto nvcc_path = get_nvcc_root_path();
    if (!nvcc_path.empty()) {
        auto cudart_header_path = nvcc_path + ".." + PATH_SPLITER + "include" +
                                  PATH_SPLITER + "cuda_runtime.h";
        //! double check path_to_nvcc/../include/cuda_runtime.h exists
        auto ret = check_file_exist(cudart_header_path.c_str(), F_OK);
        if (ret == 0) {
            paths.emplace_back(nvcc_path + "..");
            paths.emplace_back(nvcc_path + ".." + PATH_SPLITER + "include");
        }
    }

    // 3. use libcudart.so library path
    char cuda_lib_path[PATH_MAX];
    auto handle = dlopen("libcudart.so", RTLD_GLOBAL | RTLD_LAZY);
    if (handle != nullptr) {
        mgb_assert(
                dlinfo(handle, RTLD_DI_ORIGIN, cuda_lib_path) != -1, "%s", dlerror());
        paths.emplace_back(
                std::string(cuda_lib_path) + PATH_SPLITER + ".." + PATH_SPLITER +
                "include");
    }
    mgb_assert(
            paths.size() > 0,
            "can't find cuda include path, check your environment of cuda, "
            "try one of this solutions "
            "1. set CUDA_BIN_PATH to cuda home path "
            "2. add nvcc path in PATH "
            "3. add libcudart.so path in LD_LIBRARY_PATH");
    return paths;
}

#else