@@ -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
@@ -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);
@@ -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];