ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-06 15:54:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a type utility

Browse Source
This commit is contained in:
Rostyslav Geyyer
2025-03-12 16:04:24 +00:00
parent a2200a4e42
commit 489961c19a
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
include/ck/utility/data_type.hpp
View File

@@ -346,10 +346,6 @@ inline constexpr bool is_native_type()
is_same<T, f6_t>::value || is_same<T, bf6_t>::value;
}
// non_native_vector_base
template <typename T, index_t N, typename Enable = void>
struct non_native_vector_base;
// vector_type
template <typename T, index_t N, typename Enable = void>
struct vector_type;
@@ -420,13 +416,6 @@ struct scalar_type<T __attribute__((ext_vector_type(N)))>
static constexpr index_t vector_size = N;
};
template <typename T, index_t N>
struct scalar_type<non_native_vector_base<T, N>>
{
using type = T;
static constexpr index_t vector_size = N;
};
template <typename T, index_t N>
struct scalar_type<vector_type<T, N>>
{
@@ -1670,6 +1659,9 @@ struct vector_type<T, 256, typename ck::enable_if_t<is_native_type<T>()>>
}
};
template <typename T, index_t N, typename Enable = void>
struct non_native_vector_base;
template <typename T>
struct nnvb_data_t_selector
{
@@ -1902,6 +1894,14 @@ struct scalar_type<non_native_vector_base<pk_i4_t, N>>
static constexpr index_t vector_size = N;
};
template <index_t N>
struct scalar_type<non_native_vector_base<f4x2_pk_t, N>>
{
using type = typename non_native_vector_base<f4x2_pk_t, N>::data_t;
static constexpr index_t vector_size = N;
};
// non-native vector_type implementation
template <typename T>
struct vector_type<T, 1, typename ck::enable_if_t<!is_native_type<T>()>>