constrain CUDA version for cuFuncSetAttribute

leofang · leofang · commit 38fcde8462a6 · 2019-08-10T13:22:33.000-04:00
diff --git a/cupy/cuda/cupy_cuda.h b/cupy/cuda/cupy_cuda.h
@@ -22,6 +22,10 @@ extern "C" {
 
 #if CUDA_VERSION < 9000
 
+CUresult cuFuncSetAttribute(...) {
+    return CUDA_ERROR_NOT_SUPPORTED;
+}
+
 typedef enum {} cublasMath_t;
 
 cublasStatus_t cublasSetMathMode(...) {
diff --git a/cupy/cuda/driver.pyx b/cupy/cuda/driver.pyx
@@ -256,6 +256,8 @@ cpdef int funcGetAttribute(int attribute, intptr_t f):
 
 
 cpdef funcSetAttribute(intptr_t f, int attribute, int value):
+    if CUDA_VERSION < 9000:
+        raise RuntimeError("Your CUDA does not support cuFuncSetAttribute.")
     with nogil:
         status = cuFuncSetAttribute(
             <Function> f,
