diff --git a/test/gemm_universal/test_gemm_universal_wmma_fp16.cpp b/test/gemm_universal/test_gemm_universal_wmma_fp16.cpp index 4422c5589e..f249c5ca50 100644 --- a/test/gemm_universal/test_gemm_universal_wmma_fp16.cpp +++ b/test/gemm_universal/test_gemm_universal_wmma_fp16.cpp @@ -68,6 +68,7 @@ using KernelTypes_MK_NK = ::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> >; @@ -87,7 +88,6 @@ 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> >;