Keep this function public

include/starpu-util.h

@@ -228,6 +228,10 @@ void starpu_execute_on_each_worker(void (*func)(void *), void *arg, uint32_t whe
 void starpu_create_sync_task(starpu_tag_t sync_tag, unsigned ndeps, starpu_tag_t *deps,
 				void (*callback)(void *), void *callback_arg);
 
+#ifdef USE_CUDA
+cudaStream_t *starpu_get_local_cuda_stream(void);
+#endif
+
 #ifdef __cplusplus
 }
 #endif

src/drivers/cuda/driver_cuda.h

@@ -46,7 +46,6 @@ unsigned get_cuda_device_count(void);
 #ifdef USE_CUDA
 void _starpu_init_cuda(void);
 void *_starpu_cuda_worker(void *);
-cudaStream_t *starpu_get_local_cuda_stream(void);
 #endif
 
 #endif //  __DRIVER_CUDA_H__