Sfoglia il codice sorgente

Add missing prototypes

Samuel Thibault 13 anni fa
parent
commit
6af576989c
3 ha cambiato i file con 4 aggiunte e 0 eliminazioni
  1. 1 0
      include/starpu_cuda.h
  2. 1 0
      include/starpu_opencl.h
  3. 2 0
      src/datawizard/memalloc.c

+ 1 - 0
include/starpu_cuda.h

@@ -37,6 +37,7 @@ void starpu_cuda_report_error(const char *func, const char *file, int line, cuda
 #define STARPU_CUDA_REPORT_ERROR(status) \
 	starpu_cuda_report_error(__starpu_func__, __FILE__, __LINE__, status)
 
+size_t starpu_cuda_get_global_mem_size(int devid);
 cudaStream_t starpu_cuda_get_local_stream(void);
 
 #ifdef __cplusplus

+ 1 - 0
include/starpu_opencl.h

@@ -49,6 +49,7 @@ struct starpu_opencl_program {
         cl_program programs[STARPU_MAXOPENCLDEVS];
 };
 
+size_t starpu_opencl_get_global_mem_size(int devid);
 void starpu_opencl_get_context(int devid, cl_context *context);
 void starpu_opencl_get_device(int devid, cl_device_id *device);
 void starpu_opencl_get_queue(int devid, cl_command_queue *queue);

+ 2 - 0
src/datawizard/memalloc.c

@@ -17,6 +17,8 @@
 
 #include <datawizard/memalloc.h>
 #include <datawizard/footprint.h>
+#include <starpu_cuda.h>
+#include <starpu_opencl.h>
 
 /* This per-node RW-locks protect mc_list and memchunk_cache entries */
 static pthread_rwlock_t mc_rwlock[STARPU_MAXNODES];