unary.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2018-2024, 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
 *
 *     http://www.apache.org/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/fixed_point.hpp>
#include <cudf/types.hpp>
#include <cudf/utilities/default_stream.hpp>
#include <cudf/utilities/traits.hpp>
 
#include <rmm/mr/device/per_device_resource.hpp>
#include <rmm/resource_ref.hpp>
 
#include <memory>
 
namespace cudf {
template <typename Fixed,
          typename Floating,
          typename cuda::std::enable_if_t<is_fixed_point<Fixed>() &&
                                          cuda::std::is_floating_point_v<Floating>>* = nullptr>
CUDF_HOST_DEVICE Fixed convert_floating_to_fixed(Floating floating, numeric::scale_type scale)
{
  using Rep          = typename Fixed::rep;
  auto const shifted = numeric::detail::shift<Rep, Fixed::rad>(floating, scale);
  numeric::scaled_integer<Rep> scaled{static_cast<Rep>(shifted), scale};
  return Fixed(scaled);
}
 
template <typename Floating,
          typename Fixed,
          typename cuda::std::enable_if_t<cuda::std::is_floating_point_v<Floating> &&
                                          is_fixed_point<Fixed>()>* = nullptr>
CUDF_HOST_DEVICE Floating convert_fixed_to_floating(Fixed fixed)
{
  using Rep         = typename Fixed::rep;
  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,
          typename cuda::std::enable_if_t<cuda::std::is_floating_point_v<Floating>>* = nullptr>
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,         
};
 
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 = rmm::mr::get_current_device_resource());
 
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 = rmm::mr::get_current_device_resource());
 
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 = rmm::mr::get_current_device_resource());
 
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 = rmm::mr::get_current_device_resource());
 
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 = rmm::mr::get_current_device_resource());
 
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 = rmm::mr::get_current_device_resource());
  // end of group
}  // namespace cudf