dictionary.hpp

查看此文件的文档。

/*
 * 版权所有 (c) 2020-2024, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可证 2.0 版（“许可证”）获得许可；
 * 除非符合许可证的规定，否则您不得使用此文件。
 * 您可以获取许可证的副本，地址为：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则根据许可证分发的软件
 * 按“原样”提供，不附带任何明示或暗示的保证或条件。
 * 请参阅许可证，了解特定语言的权限和限制。
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#pragma once

#include <cudf/types.hpp>

#include <cuda_runtime.h>

#include <limits>

namespace CUDF_EXPORT cudf {
template <typename IndexType>
struct dictionary_wrapper {
 using value_type = IndexType;

 dictionary_wrapper() = default;
 ~dictionary_wrapper() = default;
 dictionary_wrapper(dictionary_wrapper&&) = default;
 dictionary_wrapper(dictionary_wrapper const&) = default;

 dictionary_wrapper& operator=(dictionary_wrapper&&) = default;

 dictionary_wrapper& operator=(dictionary_wrapper const&) = default;

 CUDF_HOST_DEVICE inline constexpr explicit dictionary_wrapper(value_type v) : _value{v} {}

 CUDF_HOST_DEVICE inline explicit operator value_type() const { return _value; }

 CUDF_HOST_DEVICE [[nodiscard]] inline value_type value() const { return _value; }

 static CUDF_HOST_DEVICE inline constexpr value_type max_value()
 {
 return std::numeric_limits<value_type>::max();
 }

 static CUDF_HOST_DEVICE inline constexpr value_type min_value()
 {
 return std::numeric_limits<value_type>::min();
 }

 static CUDF_HOST_DEVICE inline constexpr value_type lowest_value()
 {
 return std::numeric_limits<value_type>::lowest();
 }

 private
 value_type _value;
};

// comparison operators
template <typename Integer>
CUDF_HOST_DEVICE inline bool operator==(dictionary_wrapper<Integer> const& lhs,
 dictionary_wrapper<Integer> const& rhs)
{
 return lhs.value() == rhs.value();
}

template <typename Integer>
CUDF_HOST_DEVICE inline bool operator!=(dictionary_wrapper<Integer> const& lhs,
 dictionary_wrapper<Integer> const& rhs)
{
 return lhs.value() != rhs.value();
}

template <typename Integer>
CUDF_HOST_DEVICE inline bool operator<=(dictionary_wrapper<Integer> const& lhs,
 dictionary_wrapper<Integer> const& rhs)
{
 return lhs.value() <= rhs.value();
}

template <typename Integer>
CUDF_HOST_DEVICE inline bool operator>=(dictionary_wrapper<Integer> const& lhs,
 dictionary_wrapper<Integer> const& rhs)
{
 return lhs.value() >= rhs.value();
}

template <typename Integer>
CUDF_HOST_DEVICE inline constexpr bool operator<(dictionary_wrapper<Integer> const& lhs,
 dictionary_wrapper<Integer> const& rhs)
{
 return lhs.value() < rhs.value();
}

template <typename Integer>
CUDF_HOST_DEVICE inline bool operator>(dictionary_wrapper<Integer> const& lhs,
 dictionary_wrapper<Integer> const& rhs)
{
 return lhs.value() > rhs.value();
}

using dictionary32 = dictionary_wrapper<int32_t>;
 // end of group
} // namespace CUDF_EXPORT cudf