Explorar o código

We need to include cuda and opencl headers to use CUDA streams or OpenCL
contexts.

Cédric Augonnet %!s(int64=15) %!d(string=hai) anos
pai
achega
5539ad73fa
Modificáronse 1 ficheiros con 6 adicións e 0 borrados
  1. 6 0
      include/starpu_data_interfaces.h

+ 6 - 0
include/starpu_data_interfaces.h

@@ -19,12 +19,18 @@
 
 #include <starpu.h>
 #include <starpu_data.h>
+#include <starpu_opencl.h>
 
 #ifdef STARPU_USE_GORDON
 /* to get the gordon_strideSize_t data structure from gordon */
 #include <gordon.h>
 #endif
 
+#ifdef STARPU_USE_CUDA
+/* to use CUDA streams */
+#include <cuda_runtime.h>
+#endif
+
 #ifdef __cplusplus
 extern "C" {
 #endif