From 2db1ef6a5628ff1b358904ab6f6cfe955131296d Mon Sep 17 00:00:00 2001 From: Quentin Bourgerie Date: Tue, 6 Dec 2022 09:54:33 +0100 Subject: [PATCH] fix(cuda): Include cuda_runtime.h in device.h to include the defininition of cudaStream_t --- include/device.h | 1 + 1 file changed, 1 insertion(+) diff --git a/include/device.h b/include/device.h index 3dd11ebd6..89ef42b6d 100644 --- a/include/device.h +++ b/include/device.h @@ -1,4 +1,5 @@ #include +#include extern "C" { void *cuda_create_stream(uint32_t gpu_index);