mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-17 03:19:48 +00:00
82ad74304e
* delete obselete files
* move files
* build
* update cmake
* update cmake
* fix build
* reorg examples
* update cmake for example and test
[ROCm/composable_kernel commit: 5d37d7bff4]
51 lines
1.5 KiB
C++
51 lines
1.5 KiB
C++
#ifndef CK_INTEGRAL_CONSTANT_HPP
|
|
#define CK_INTEGRAL_CONSTANT_HPP
|
|
|
|
namespace ck {
|
|
|
|
template <class T, T v>
|
|
struct integral_constant
|
|
{
|
|
static constexpr T value = v;
|
|
typedef T value_type;
|
|
typedef integral_constant type;
|
|
__host__ __device__ constexpr operator value_type() const noexcept { return value; }
|
|
__host__ __device__ constexpr value_type operator()() const noexcept { return value; }
|
|
};
|
|
|
|
template <typename TX, TX X, typename TY, TY Y>
|
|
__host__ __device__ constexpr auto operator+(integral_constant<TX, X>, integral_constant<TY, Y>)
|
|
{
|
|
return integral_constant<decltype(X + Y), X + Y>{};
|
|
}
|
|
|
|
template <typename TX, TX X, typename TY, TY Y>
|
|
__host__ __device__ constexpr auto operator-(integral_constant<TX, X>, integral_constant<TY, Y>)
|
|
{
|
|
static_assert(Y <= X, "wrong!");
|
|
return integral_constant<decltype(X - Y), X - Y>{};
|
|
}
|
|
|
|
template <typename TX, TX X, typename TY, TY Y>
|
|
__host__ __device__ constexpr auto operator*(integral_constant<TX, X>, integral_constant<TY, Y>)
|
|
{
|
|
return integral_constant<decltype(X * Y), X * Y>{};
|
|
}
|
|
|
|
template <typename TX, TX X, typename TY, TY Y>
|
|
__host__ __device__ constexpr auto operator/(integral_constant<TX, X>, integral_constant<TY, Y>)
|
|
{
|
|
static_assert(Y > 0, "wrong!");
|
|
return integral_constant<decltype(X / Y), X / Y>{};
|
|
}
|
|
|
|
template <typename TX, TX X, typename TY, TY Y>
|
|
__host__ __device__ constexpr auto operator%(integral_constant<TX, X>, integral_constant<TY, Y>)
|
|
{
|
|
static_assert(Y > 0, "wrong!");
|
|
return integral_constant<decltype(X % Y), X % Y>{};
|
|
}
|
|
|
|
} // namespace ck
|
|
#endif
|