traits.cuh

前往此文件的文档。

/*
 * 版权所有 (c) 2021-2024, NVIDIA CORPORATION。
 *
 * 根据 Apache 许可证 2.0 版（“许可证”）获得许可；
 * 除非遵守许可证，否则您不得使用此文件。
 * 您可以获得许可证的副本，地址是
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则根据许可证分发的软件按“原样”分发，
 * 不附带任何明示或暗示的保证或条件。
 * 请参阅许可证中关于特定语言的权限和限制的说明。
 *
 * limitations under the License.
 */

#pragma once

#include <cudf/types.hpp>
#include <cudf/utilities/type_dispatcher.hpp>

#include <cuda/std/atomic>

namespace CUDF_EXPORT cudf {

/** \brief 指示类型 T 是否支持原子操作。对于内置类型通常为 true。 *//** @ingroup utility_types */
template <typename T>
constexpr inline bool has_atomic_support()
{
 return cuda::std::atomic<T>::is_always_lock_free;
}

struct has_atomic_support_impl {
 template <typename T>
 constexpr bool operator()()
 {
 return has_atomic_support<T>();
 }
};
/** \brief 指示类型是否支持原子操作。 *//** @ingroup utility_types */
constexpr inline bool has_atomic_support(data_type type)
{
 return cudf::type_dispatcher(type, has_atomic_support_impl{});
}
} // namespace CUDF_EXPORT cudf