Home Explore Help
Sign In
exa2pro
/
starpu-max
Watch 19
Star 0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

Define CUBLAS functions

Cédric Augonnet 14 years ago
parent
0828950dd7
commit
c0be233412
1 changed files with 4 additions and 1 deletions
Unified View Show Diff Stats
  1. 4 1
      examples/axpy/axpy.c

+ 4 - 1
examples/axpy/axpy.c
View File 

@@ -20,9 +20,12 @@
 
 #include <assert.h>
 
 #include <assert.h>
 
 #include <sys/time.h>
 
 #include <sys/time.h>
 
 
 
-
 
 #include <common/blas.h>
 
 #include <common/blas.h>
 
 
 
+#ifdef STARPU_USE_CUDA
 
+#include <cublas.h>
 
+#endif
 
+
 
 #define TYPE	float
 
 #define TYPE	float
 
 #define AXPY	SAXPY
 
 #define AXPY	SAXPY
 
 #define CUBLASAXPY	cublasSaxpy
 
 #define CUBLASAXPY	cublasSaxpy