exclude cublas

Author: js1010

.travis.yml

@@ -29,7 +29,7 @@ before_install:
   - wget https://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/7fa2af80.pub
   - sudo apt-key add 7fa2af80.pub
   - sudo apt update -qq
-  - sudo apt install -y cuda-${CUDA_SHORT/./-}
+  - sudo apt install -y cuda-core-${CUDA_SHORT/./-} cuda-cudart-dev-${CUDA_SHORT/./-} cuda-cufft-dev-${CUDA_SHORT/./-} 
   - sudo apt clean
   - export CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
   - export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}

cpp/include/utils/cuda_utils_kernels.cuh

@@ -5,7 +5,7 @@
 // LICENSE file in the root directory of this source tree.
 #pragma once
 #include <unistd.h>
-#include <cublas_v2.h>
+// #include <cublas_v2.h>
 #include <cuda_runtime.h>
 
 #include <thrust/copy.h>
@@ -39,29 +39,29 @@ inline void checkCuda(cudaError_t code, const char *file, int line) {
   }
 }
 
-inline const char* cublasGetErrorString(cublasStatus_t status) {
-  switch (status) {
-    case CUBLAS_STATUS_SUCCESS: return "CUBLAS_STATUS_SUCCESS";
-    case CUBLAS_STATUS_NOT_INITIALIZED: return "CUBLAS_STATUS_NOT_INITIALIZED";
-    case CUBLAS_STATUS_ALLOC_FAILED: return "CUBLAS_STATUS_ALLOC_FAILED";
-    case CUBLAS_STATUS_INVALID_VALUE: return "CUBLAS_STATUS_INVALID_VALUE";
-    case CUBLAS_STATUS_ARCH_MISMATCH: return "CUBLAS_STATUS_ARCH_MISMATCH";
-    case CUBLAS_STATUS_MAPPING_ERROR: return "CUBLAS_STATUS_MAPPING_ERROR";
-    case CUBLAS_STATUS_EXECUTION_FAILED: return "CUBLAS_STATUS_EXECUTION_FAILED";
-    case CUBLAS_STATUS_INTERNAL_ERROR: return "CUBLAS_STATUS_INTERNAL_ERROR";
-  }
-  return "Unknown";
-}
-
-#define CHECK_CUBLAS(code) { checkCublas((code), __FILE__, __LINE__); }
-inline void checkCublas(cublasStatus_t code, const char * file, int line) {
-  if (code != CUBLAS_STATUS_SUCCESS) {
-    std::stringstream err;
-    err << "cublas error: " << cublasGetErrorString(code)
-        << " (" << file << ":" << line << ")";
-    throw std::runtime_error(err.str());
-  }
-}
+// inline const char* cublasGetErrorString(cublasStatus_t status) {
+//   switch (status) {
+//     case CUBLAS_STATUS_SUCCESS: return "CUBLAS_STATUS_SUCCESS";
+//     case CUBLAS_STATUS_NOT_INITIALIZED: return "CUBLAS_STATUS_NOT_INITIALIZED";
+//     case CUBLAS_STATUS_ALLOC_FAILED: return "CUBLAS_STATUS_ALLOC_FAILED";
+//     case CUBLAS_STATUS_INVALID_VALUE: return "CUBLAS_STATUS_INVALID_VALUE";
+//     case CUBLAS_STATUS_ARCH_MISMATCH: return "CUBLAS_STATUS_ARCH_MISMATCH";
+//     case CUBLAS_STATUS_MAPPING_ERROR: return "CUBLAS_STATUS_MAPPING_ERROR";
+//     case CUBLAS_STATUS_EXECUTION_FAILED: return "CUBLAS_STATUS_EXECUTION_FAILED";
+//     case CUBLAS_STATUS_INTERNAL_ERROR: return "CUBLAS_STATUS_INTERNAL_ERROR";
+//   }
+//   return "Unknown";
+// }
+// 
+// #define CHECK_CUBLAS(code) { checkCublas((code), __FILE__, __LINE__); }
+// inline void checkCublas(cublasStatus_t code, const char * file, int line) {
+//   if (code != CUBLAS_STATUS_SUCCESS) {
+//     std::stringstream err;
+//     err << "cublas error: " << cublasGetErrorString(code)
+//         << " (" << file << ":" << line << ")";
+//     throw std::runtime_error(err.str());
+//   }
+// }
 
 inline DeviceInfo GetDeviceInfo() {
   DeviceInfo ret;

setup.py

@@ -92,7 +92,7 @@ def __init__(self, name):
             extra_compile_args=extra_compile_args,
             extra_link_args=["-fopenmp"],
             library_dirs=[CUDA['lib64']],
-            libraries=['cudart', 'cublas', 'curand'],
+            libraries=['cudart', 'curand'],
             extra_objects=[],
             include_dirs=[ \
               "cpp/include/", np.get_include(), pybind11.get_include(),
@@ -107,7 +107,7 @@ def __init__(self, name):
             extra_compile_args=extra_compile_args,
             extra_link_args=["-fopenmp"],
             library_dirs=[CUDA['lib64']],
-            libraries=['cudart', 'cublas', 'curand'],
+            libraries=['cudart', 'curand'],
             extra_objects=[],
             include_dirs=[ \
               "cpp/include/", np.get_include(), pybind11.get_include(),