made udf transform benchmark run on provided stream

rapidsai · Jan 9, 2025 · 578a56f · 578a56f
1 parent bc1528d
commit 578a56f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cpp/benchmarks/transform/polynomials.cpp b/cpp/benchmarks/transform/polynomials.cpp
@@ -78,7 +78,11 @@ __device__ inline void    fdsf   (
 }
 )***";
 
-    cudf::transform(column_view, udf, cudf::data_type{cudf::type_to_id<key_type>()}, false);
+    cudf::transform(column_view,
+                    udf,
+                    cudf::data_type{cudf::type_to_id<key_type>()},
+                    false,
+                    launch.get_stream().get_stream());
   });
 }