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

Updated the set of tests for FP16

Browse Source
This commit is contained in:
Anca Hamuraru
2025-05-20 13:16:48 +00:00
parent 0a5e6d48ec
commit ae21582a88
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/gemm_universal/test_gemm_universal_wmma_fp16.cpp
View File

@@ -58,6 +58,7 @@ using KernelTypes_MK_KN = ::testing::Types<
// ADataType, BDataType, ComputeDataType, CDataType
#if defined(CK_ENABLE_FP8)
std::tuple< F8, F16, F16, F16>,
std::tuple< F16, F8, F16, F16>,
#endif
std::tuple< F16, F16, F16, F16>
>;
@@ -66,6 +67,7 @@ using KernelTypes_MK_NK = ::testing::Types<
// ADataType, BDataType, ComputeDataType, CDataType
#if defined(CK_ENABLE_FP8)
std::tuple< F8, F16, F16, F16>,
std::tuple< F16, F8, F16, F16>,
#endif
std::tuple< F16, F16, F16, F16>
>;
@@ -85,6 +87,7 @@ using KernelTypes_KM_KN = ::testing::Types<
#if defined(CK_ENABLE_FP8)
std::tuple< F8, F16, F16, F16>,
std::tuple< F16, F8, F16, F16>,
std::tuple< F16, I4, F16, F16>,
#endif
std::tuple< F16, F16, F16, F16>
>;