Browse Source

fix build with older cuda

Samuel Thibault 8 years ago
parent
commit
96bca6e25e
2 changed files with 5 additions and 0 deletions
  1. 1 0
      configure.ac
  2. 4 0
      src/drivers/cuda/starpu_cusparse.c

+ 1 - 0
configure.ac

@@ -1118,6 +1118,7 @@ if test x$enable_cuda = xyes; then
 	AC_CHECK_HEADERS([cuda_gl_interop.h])
 
 	AC_CHECK_LIB([cusparse], [cusparseCreate])
+	AC_CHECK_DECLS([cusparseSetStream], [], [], [[#include <cusparse.h>]])
 fi
 
 dnl Hey dude, are you around?

+ 4 - 0
src/drivers/cuda/starpu_cusparse.c

@@ -30,7 +30,11 @@ static cusparseHandle_t main_handle;
 static void init_cusparse_func(void *args STARPU_ATTRIBUTE_UNUSED)
 {
 	cusparseCreate(&cusparse_handles[starpu_worker_get_id_check()]);
+#if HAVE_DECL_CUSPARSESETSTREAM
 	cusparseSetStream(cusparse_handles[starpu_worker_get_id_check()], starpu_cuda_get_local_stream());
+#else
+	cusparseSetKernelStream(cusparse_handles[starpu_worker_get_id_check()], starpu_cuda_get_local_stream());
+#endif
 }
 
 static void shutdown_cusparse_func(void *args STARPU_ATTRIBUTE_UNUSED)