Browse Source

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

Cédric Augonnet 15 years ago
parent
commit
5539ad73fa
1 changed files with 6 additions and 0 deletions
  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