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

Unify saturates<> implementation

Browse Source
This commit is contained in:
Po Yen Chen
2024-04-09 12:49:07 +00:00
parent ecc64bce12
commit a66409cfd9
1 changed files with 3 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
include/ck_tile/core/utility/unary_element_function.hpp
View File

@@ -56,20 +56,9 @@ struct saturates
CK_TILE_HOST_DEVICE constexpr auto operator()(const From& from) const
-> std::enable_if_t<std::is_arithmetic_v<From>, From>
{
if constexpr(std::is_floating_point_v<To> || std::is_same_v<To, half_t> ||
std::is_same_v<To, bfloat16_t> || std::is_same_v<To, fp8_t> ||
std::is_same_v<To, bf8_t>)
{
return clamp(from,
type_convert<From>(numeric<To>::lowest()),
type_convert<From>(numeric<To>::max()));
}
else
{
return clamp(from,
type_convert<From>(numeric<To>::min()),
type_convert<From>(numeric<To>::max()));
}
return clamp(from,
type_convert<From>(numeric<To>::lowest()),
type_convert<From>(numeric<To>::max()));
}
};