made binaryop benchmark run on provided stream

rapidsai · Jan 9, 2025 · 349e8d3 · 349e8d3
1 parent 578a56f
commit 349e8d3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cpp/benchmarks/binaryop/polynomials.cpp b/cpp/benchmarks/binaryop/polynomials.cpp
@@ -58,7 +58,8 @@ static void BM_binaryop_polynomials(nvbench::state& state)
   state.exec(nvbench::exec_tag::sync, [&](nvbench::launch& launch) {
     // computes polynomials: (((ax + b)x + c)x + d)x + e... = ax**4 + bx**3 + cx**2 + dx + e....
 
-    auto result = cudf::make_column_from_scalar(constants[0], num_rows);
+    auto result =
+      cudf::make_column_from_scalar(constants[0], num_rows, launch.get_stream().get_stream());
 
     for (cudf::size_type i = 0; i < order; i++) {
       auto product = cudf::binary_operation(result->view(),