unary.hpp

前往此文件的文档。

/*
 * Copyright (c) 2018-2025, NVIDIA CORPORATION.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#pragma once

#include <cudf/fixed_point/detail/floating_conversion.hpp>
#include <cudf/fixed_point/fixed_point.hpp>
#include <cudf/types.hpp>
#include <cudf/utilities/default_stream.hpp>
#include <cudf/utilities/export.hpp>
#include <cudf/utilities/memory_resource.hpp>
#include <cudf/utilities/traits.hpp>

#include <memory>

namespace CUDF_EXPORT cudf {
template <typename Fixed,
 typename Floating,
 CUDF_ENABLE_IF(cuda::std::is_floating_point_v<Floating>&& is_fixed_point<Fixed>())>
CUDF_HOST_DEVICE Fixed convert_floating_to_fixed(Floating floating, numeric::scale_type scale)
{
 using Rep = typename Fixed::rep;
 auto const value = [&]() {
 if constexpr (Fixed::rad == numeric::Radix::BASE_10) {
 return numeric::detail::convert_floating_to_integral<Rep>(floating, scale);
 } else {
 return static_cast<Rep>(numeric::detail::shift<Rep, Fixed::rad>(floating, scale));
 }
 }();

 return Fixed(numeric::scaled_integer<Rep>{value, scale});
}

template <typename Floating,
 typename Fixed,
 CUDF_ENABLE_IF(cuda::std::is_floating_point_v<Floating>&& is_fixed_point<Fixed>())>
CUDF_HOST_DEVICE Floating convert_fixed_to_floating(Fixed fixed)
{
 using Rep = typename Fixed::rep;
 if constexpr (Fixed::rad == numeric::Radix::BASE_10) {
 return numeric::detail::convert_integral_to_floating<Floating>(fixed.value(), fixed.scale());
 } else {
 auto const casted = static_cast<Floating>(fixed.value());
 auto const scale = numeric::scale_type{-fixed.scale()};
 return numeric::detail::shift<Rep, Fixed::rad>(casted, scale);
 }
}

template <typename Floating,
 typename Input,
 CUDF_ENABLE_IF(cuda::std::is_floating_point_v<Floating>)>
CUDF_HOST_DEVICE Floating convert_to_floating(Input input)
{
 if constexpr (is_fixed_point<Input>()) {
 return convert_fixed_to_floating<Floating>(input);
 } else {
 return static_cast<Floating>(input);
 }
}

enum class unary_operator : int32_t {
 SIN,
 COS,
 TAN,
 ARCSIN,
 ARCCOS,
 ARCTAN,
 SINH,
 COSH,
 TANH,
 ARCSINH,
 ARCCOSH,
 ARCTANH,
 EXP,
 LOG,
 SQRT,
 CBRT,
 CEIL,
 FLOOR,
 ABS,
 RINT,
 BIT_INVERT,
 NOT,
 NEGATE,
};

std::unique_ptr<cudf::column> unary_operation(
 cudf::column_view const& input,
 cudf::unary_operator op,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<cudf::column> is_null(
 cudf::column_view const& input,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<cudf::column> is_valid(
 cudf::column_view const& input,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<column> cast(
 column_view const& input,
 data_type out_type,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

bool is_supported_cast(data_type from, data_type to) noexcept;

std::unique_ptr<column> is_nan(
 cudf::column_view const& input,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<column> is_not_nan(
 cudf::column_view const& input,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
 // end of group
} // namespace CUDF_EXPORT cudf