switched to using rmm also inside algorithm

karthikeyann · Jul 16, 2022 · b260610 · b260610
1 parent 9b20d16
commit b260610
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cpp/tests/io/fst/logical_stack_test.cu b/cpp/tests/io/fst/logical_stack_test.cu
@@ -23,6 +23,7 @@
 #include <rmm/cuda_stream.hpp>
 #include <rmm/cuda_stream_view.hpp>
 #include <rmm/device_uvector.hpp>
+#include <rmm/device_buffer.hpp>
 
 #include <cstdlib>
 #include <iostream>