fix(cuda): Include cuda_runtime.h in device.h to include the defininition of cudaStream_t

This commit is contained in:
Quentin Bourgerie
2022-12-06 09:54:33 +01:00
committed by Agnès Leroy
parent 0aedb1a4f4
commit 2db1ef6a56

View File

@@ -1,4 +1,5 @@
#include <cstdint>
#include <cuda_runtime.h>
extern "C" {
void *cuda_create_stream(uint32_t gpu_index);