Make it buildable with CUDA 8, CUDA 7.5 or No CUDA
@@ -62,8 +62,29 @@ cudaError_t cudaMemAdvise(const void *devPtr, size_t count,
return cudaErrorUnknown;
}
+typedef enum {} cublasGemmAlgo_t;
+
+cublasStatus_t cublasGemmEx(...) {
+ return CUBLAS_STATUS_NOT_SUPPORTED;
+}
+
#endif // #if CUDA_VERSION < 8000
+
+#if CUDA_VERSION < 9000
+
+typedef enum {} cublasMath_t;
+
+cublasStatus_t cublasSetMathMode(...) {
+ return CUBLAS_STATUS_NOT_SUPPORTED;
+}
+
+cublasStatus_t cublasGetMathMode(...) {
+ return CUBLAS_STATUS_NOT_SUPPORTED;
+}
+
+#endif // #if CUDA_VERSION < 9000
+
} // extern "C"
#else // #ifndef CUPY_NO_CUDA
@@ -548,6 +569,10 @@ cublasStatus_t cublasSgemmEx(...) {
return CUBLAS_STATUS_SUCCESS;
}
+cublasStatus_t cublasGemmEx(...) {
+ return CUBLAS_STATUS_SUCCESS;
+}
+
cublasStatus_t cublasStrsm(...) {
return CUBLAS_STATUS_SUCCESS;
}
