conv.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
#include <cudf/fixed_point/detail/floating_conversion.hpp>
#include <cudf/fixed_point/fixed_point.hpp>
#include <cudf/types.hpp>
#include <cudf/utilities/export.hpp>
#include <cudf/utilities/traits.hpp>
 
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);
  }
}
  // end of group
}  // namespace CUDF_EXPORT cudf