diff --git a/src/main/cpp/src/decimal_utils.hpp b/src/main/cpp/src/decimal_utils.hpp index 1011a0a574..95c6c56c3d 100644 --- a/src/main/cpp/src/decimal_utils.hpp +++ b/src/main/cpp/src/decimal_utils.hpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2022, NVIDIA CORPORATION. + * Copyright (c) 2022-2023, NVIDIA CORPORATION. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -14,8 +14,12 @@ * limitations under the License. */ +#pragma once + #include #include +#include + #include #include @@ -26,35 +30,35 @@ std::unique_ptr multiply_decimal128( cudf::column_view const& a, cudf::column_view const& b, int32_t product_scale, - rmm::cuda_stream_view stream = rmm::cuda_stream_default); + rmm::cuda_stream_view stream = cudf::get_default_stream()); std::unique_ptr divide_decimal128( cudf::column_view const& a, cudf::column_view const& b, int32_t quotient_scale, - rmm::cuda_stream_view stream = rmm::cuda_stream_default); + rmm::cuda_stream_view stream = cudf::get_default_stream()); std::unique_ptr integer_divide_decimal128( cudf::column_view const& a, cudf::column_view const& b, int32_t quotient_scale, - rmm::cuda_stream_view stream = rmm::cuda_stream_default); + rmm::cuda_stream_view stream = cudf::get_default_stream()); std::unique_ptr remainder_decimal128( cudf::column_view const& a, cudf::column_view const& b, int32_t remainder_scale, - rmm::cuda_stream_view stream = rmm::cuda_stream_default); + rmm::cuda_stream_view stream = cudf::get_default_stream()); std::unique_ptr add_decimal128( cudf::column_view const& a, cudf::column_view const& b, int32_t quotient_scale, - rmm::cuda_stream_view stream = rmm::cuda_stream_default); + rmm::cuda_stream_view stream = cudf::get_default_stream()); std::unique_ptr sub_decimal128( cudf::column_view const& a, cudf::column_view const& b, int32_t quotient_scale, - rmm::cuda_stream_view stream = rmm::cuda_stream_default); + rmm::cuda_stream_view stream = cudf::get_default_stream()); } // namespace cudf::jni diff --git a/src/main/cpp/src/parse_uri.hpp b/src/main/cpp/src/parse_uri.hpp index 3237e347ab..c65d06d80a 100644 --- a/src/main/cpp/src/parse_uri.hpp +++ b/src/main/cpp/src/parse_uri.hpp @@ -18,6 +18,7 @@ #include #include +#include #include @@ -35,7 +36,7 @@ namespace spark_rapids_jni { */ std::unique_ptr parse_uri_to_protocol( cudf::strings_column_view const& input, - rmm::cuda_stream_view stream = rmm::cuda_stream_default, + rmm::cuda_stream_view stream = cudf::get_default_stream(), rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource()); } // namespace spark_rapids_jni diff --git a/src/main/cpp/src/row_conversion.hpp b/src/main/cpp/src/row_conversion.hpp index 84ee729d55..6e9835e3d2 100644 --- a/src/main/cpp/src/row_conversion.hpp +++ b/src/main/cpp/src/row_conversion.hpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2022, NVIDIA CORPORATION. + * Copyright (c) 2022-2023, NVIDIA CORPORATION. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -16,36 +16,38 @@ #pragma once -#include - #include #include +#include + #include +#include + namespace spark_rapids_jni { std::vector> convert_to_rows_fixed_width_optimized( cudf::table_view const& tbl, // TODO need something for validity - rmm::cuda_stream_view stream = rmm::cuda_stream_default, + rmm::cuda_stream_view stream = cudf::get_default_stream(), rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource()); std::vector> convert_to_rows( cudf::table_view const& tbl, // TODO need something for validity - rmm::cuda_stream_view stream = rmm::cuda_stream_default, + rmm::cuda_stream_view stream = cudf::get_default_stream(), rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource()); std::unique_ptr convert_from_rows_fixed_width_optimized( cudf::lists_column_view const& input, std::vector const& schema, - rmm::cuda_stream_view stream = rmm::cuda_stream_default, + rmm::cuda_stream_view stream = cudf::get_default_stream(), rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource()); std::unique_ptr convert_from_rows( cudf::lists_column_view const& input, std::vector const& schema, - rmm::cuda_stream_view stream = rmm::cuda_stream_default, + rmm::cuda_stream_view stream = cudf::get_default_stream(), rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource()); } // namespace spark_rapids_jni