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

Add test case for bf16_i4

Browse Source
This commit is contained in:
Anton Gorenko
2025-05-21 18:02:00 +05:00
parent 02bf56a6de
commit dd7ac95aff
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/gemm_universal/test_gemm_universal_wmma_bf16.cpp
View File

@@ -10,7 +10,6 @@
using I4 = ck::pk_i4_t;
using BF16 = ck::bhalf_t;
using F32 = float;
using F16 = ck::half_t;
using Row = ck::tensor_layout::gemm::RowMajor;
using Col = ck::tensor_layout::gemm::ColumnMajor;
@@ -61,7 +60,7 @@ using KernelTypes_MK_KN = ::testing::Types<
using KernelTypes_MK_NK = ::testing::Types<
// ADataType, BDataType, ComputeDataType, CDataType
#if defined(CK_ENABLE_FP8)
std::tuple< BF16, I4, BF16, F16>,
std::tuple< BF16, I4, BF16, BF16>,
#endif
std::tuple< BF16, BF16, BF16, BF16>
>;
@@ -73,6 +72,9 @@ using KernelTypes_KM_KN = ::testing::Types<
using KernelTypes_KM_NK = ::testing::Types<
// ADataType, BDataType, ComputeDataType, CDataType
#if defined(CK_ENABLE_FP8)
std::tuple< BF16, I4, BF16, BF16>,
#endif
std::tuple< BF16, BF16, BF16, BF16>
>;
// clang-format on

2
test/gemm_universal/test_gemm_universal_wmma_fp16.cpp
View File

@@ -89,7 +89,7 @@ using KernelTypes_KM_KN = ::testing::Types<
std::tuple< F8, F16, F16, F16>,
std::tuple< F16, F8, F16, F16>,
#endif
std::tuple< F16, F16, F16, F16>
std::tuple< F16, F16, F16, F16>
>;
// clang-format on