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 <cstdint>
+#include <cuda_runtime.h>
 
 extern "C" {
 void *cuda_create_stream(uint32_t gpu_index);