From dd7ac95afff4f6c19be023738db8195ccf9dafda Mon Sep 17 00:00:00 2001 From: Anton Gorenko Date: Wed, 21 May 2025 18:02:00 +0500 Subject: [PATCH] Add test case for bf16_i4 --- test/gemm_universal/test_gemm_universal_wmma_bf16.cpp | 6 ++++-- test/gemm_universal/test_gemm_universal_wmma_fp16.cpp | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/test/gemm_universal/test_gemm_universal_wmma_bf16.cpp b/test/gemm_universal/test_gemm_universal_wmma_bf16.cpp index 941580c360..f8c8d6efb1 100644 --- a/test/gemm_universal/test_gemm_universal_wmma_bf16.cpp +++ b/test/gemm_universal/test_gemm_universal_wmma_bf16.cpp @@ -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 diff --git a/test/gemm_universal/test_gemm_universal_wmma_fp16.cpp b/test/gemm_universal/test_gemm_universal_wmma_fp16.cpp index f249c5ca50..fa0220aa34 100644 --- a/test/gemm_universal/test_gemm_universal_wmma_fp16.cpp +++ b/test/gemm_universal/test_gemm_universal_wmma_fp16.cpp @@ -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