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

Simplify static_cast if-lands (#1828)

Browse Source
This commit is contained in:
Mateusz Ozga
2025-01-21 23:23:19 +01:00
committed by GitHub
parent 3c93d3c444
commit 3db77bc4f2
3 changed files with 63 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
include/ck_tile/core/utility/type_traits.hpp
View File

@@ -109,4 +109,22 @@ CK_TILE_HOST_DEVICE PY c_style_pointer_cast(PX p_x)
#pragma clang diagnostic pop
}
template <typename CompareTo, typename... Rest>
struct is_any_of : std::false_type
{
};
template <typename CompareTo, typename FirstType>
struct is_any_of<CompareTo, FirstType> : std::is_same<CompareTo, FirstType>
{
};
template <typename CompareTo, typename FirstType, typename... Rest>
struct is_any_of<CompareTo, FirstType, Rest...>
: std::integral_constant<bool,
std::is_same<CompareTo, FirstType>::value ||
is_any_of<CompareTo, Rest...>::value>
{
};
} // namespace ck_tile