Add more const

rapidsai · Jul 17, 2024 · f635b71 · f635b71
1 parent 3971e4c
commit f635b71
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/cpp/benchmarks/common/cudf_datagen/dbgen.cu b/cpp/benchmarks/common/cudf_datagen/dbgen.cu
@@ -326,8 +326,12 @@ void generate_orders(int64_t scale_factor)
  * @brief Generate the `partsupp` table
  *
  * @param scale_factor The scale factor to use
+ * @param stream The CUDA stream to use
+ * @param mr The memory resource to use
  */
-void generate_partsupp(int64_t scale_factor)
+void generate_partsupp(int64_t const& scale_factor,
+                       rmm::cuda_stream_view stream      = cudf::get_default_stream(),
+                       rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource())
 {
   cudf::size_type num_rows_part = 200000 * scale_factor;
   cudf::size_type num_rows      = 800000 * scale_factor;
@@ -419,7 +423,7 @@ std::unique_ptr<cudf::column> calc_p_retailprice(cudf::column_view const& p_part
  * @param stream The CUDA stream to use
  * @param mr The memory resource to use
  */
-void generate_part(int64_t scale_factor,
+void generate_part(int64_t const& scale_factor,
                    rmm::cuda_stream_view stream      = cudf::get_default_stream(),
                    rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource())
 {
@@ -635,7 +639,7 @@ void generate_customer(int64_t const& scale_factor,
  * @param stream The CUDA stream to use
  * @param mr The memory resource to use
  */
-void generate_supplier(int64_t scale_factor,
+void generate_supplier(int64_t const& scale_factor,
                        rmm::cuda_stream_view stream      = cudf::get_default_stream(),
                        rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource())
 {