Headers that use std::thread now include <thread>

Fixes rapidsai#937
robertmaynard · Jan 5, 2022 · febd7f0 · febd7f0
1 parent 846d638
commit febd7f0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/include/rmm/mr/device/arena_memory_resource.hpp b/include/rmm/mr/device/arena_memory_resource.hpp
@@ -28,6 +28,7 @@
 #include <cstddef>
 #include <map>
 #include <shared_mutex>
+#include <thread>
 
 namespace rmm::mr {
 

diff --git a/tests/mr/device/arena_mr_tests.cpp b/tests/mr/device/arena_mr_tests.cpp
@@ -28,6 +28,9 @@
 
 #include <sys/stat.h>
 
+#include <thread>
+#include <vector>
+
 namespace rmm::test {
 
 namespace {