From b260610a7c7e526b21e91f35e47491fa02649aeb Mon Sep 17 00:00:00 2001
From: Elias Stehle <3958403+elstehle@users.noreply.github.com>
Date: Thu, 31 Mar 2022 05:28:17 -0700
Subject: [PATCH] switched to using rmm also inside algorithm

---
 cpp/tests/io/fst/logical_stack_test.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cpp/tests/io/fst/logical_stack_test.cu b/cpp/tests/io/fst/logical_stack_test.cu
index d76ff16f85d..aaa6200799a 100644
--- 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>