diff --git a/test/gemm_universal/test_gemm_universal_wmma_fp16.cpp b/test/gemm_universal/test_gemm_universal_wmma_fp16.cpp index 734dfe72d7..4422c5589e 100644 --- a/test/gemm_universal/test_gemm_universal_wmma_fp16.cpp +++ b/test/gemm_universal/test_gemm_universal_wmma_fp16.cpp @@ -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> >;