ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 03:37:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add functionality and tests for fp16 x fp8 and fp8 x fp16

Browse Source
This commit is contained in:
Sami Aario
2025-11-12 15:09:01 +00:00
parent f8c4868a59
commit 3a792017fb
2 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
include/ck_tile/ops/common/determine_warp_prec_type.hpp
View File

@@ -52,4 +52,18 @@ struct DetermineWarpPrecType<ck_tile::bf16_t, ck_tile::fp8_t>
{
using prec_type = float;
};
// For fp8 x fp16 or fp16 x fp8, convert fp8 to float
template <>
struct DetermineWarpPrecType<ck_tile::fp8_t, ck_tile::half_t>
{
using prec_type = float;
};
// For fp8 x fp16 or fp16 x fp8, convert fp16 to float
template <>
struct DetermineWarpPrecType<ck_tile::half_t, ck_tile::fp8_t>
{
using prec_type = float;
};
}; // namespace ck_tile