fixes sg-count & uses rmm stream in fst tests

karthikeyann · Jul 16, 2022 · 6d3eff2 · 6d3eff2
1 parent 6548836
commit 6d3eff2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cpp/tests/io/fst/fst_test.cu b/cpp/tests/io/fst/fst_test.cu
@@ -26,6 +26,7 @@
 
 #include <rmm/cuda_stream.hpp>
 #include <rmm/cuda_stream_view.hpp>
+#include <rmm/cuda_stream.hpp>
 #include <rmm/device_buffer.hpp>
 #include <rmm/device_uvector.hpp>
 
@@ -210,6 +211,8 @@ TEST_F(FstTest, GroundTruth)
   auto& d_input = static_cast<cudf::scalar_type_t<std::string>&>(*d_input_string);
   input         = d_input.to_string(stream);
 
+
+
   // Prepare input & output buffers
   constexpr std::size_t single_item = 1;
   hostdevice_vector<SymbolT> output_gpu(input.size(), stream_view);