Pārlūkot izejas kodu

Define CUBLAS functions

Cédric Augonnet 14 gadi atpakaļ
vecāks
revīzija
c0be233412
1 mainītis faili ar 4 papildinājumiem un 1 dzēšanām
  1. 4 1
      examples/axpy/axpy.c

+ 4 - 1
examples/axpy/axpy.c

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