From 6d3eff29ef386663e8feb3a103eb9795c7c7c66d Mon Sep 17 00:00:00 2001
From: Elias Stehle <3958403+elstehle@users.noreply.github.com>
Date: Thu, 2 Jun 2022 05:19:53 -0700
Subject: [PATCH] fixes sg-count & uses rmm stream in fst tests

---
 cpp/tests/io/fst/fst_test.cu | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/cpp/tests/io/fst/fst_test.cu b/cpp/tests/io/fst/fst_test.cu
index 5d169cd9ac1..296176e48f9 100644
--- 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);