瀏覽代碼

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

Cédric Augonnet 15 年之前
父節點
當前提交
5539ad73fa
共有 1 個文件被更改,包括 6 次插入0 次删除
  1. 6 0
      include/starpu_data_interfaces.h

+ 6 - 0
include/starpu_data_interfaces.h

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