diff --git a/test/gemm_add/test_gemm_multiply_multiply_wmma.cpp b/test/gemm_add/test_gemm_multiply_multiply_wmma.cpp index f17a8f975b..7a67916a23 100644 --- a/test/gemm_add/test_gemm_multiply_multiply_wmma.cpp +++ b/test/gemm_add/test_gemm_multiply_multiply_wmma.cpp @@ -49,7 +49,7 @@ public: void Run() { std::vector> lengths = { - {16, 32, 64}, {2048, 4096, 8192}, {2048, 4096, 128}}; + {16, 32, 64}, {512, 2048, 4096}, {2048, 1024, 16}}; bool all_success = true; @@ -66,18 +66,20 @@ public: all_success = all_success & - ProfileGemmMultiplyMultiplyImpl(1, 1, false, false, M, N, K, StrideA, StrideB, StrideD0, StrideD1, StrideE, 1, 1, 1, 0); + ProfileGemmMultiplyMultiplyImpl(1, 1, false, true, M, N, K, StrideA, StrideB, StrideD0, StrideD1, StrideE, 1, 1, 1, 0); } EXPECT_TRUE(all_success); } }; -using KernelTypes = - ::testing::Types, - std::tuple, - std::tuple, - std::tuple>; +using KernelTypes = ::testing::Types< +#ifdef CK_USE_WMMA_FP8 + std::tuple, + std::tuple, +#endif + std::tuple, + std::tuple>; TYPED_TEST_SUITE(TestGemmMultiplyMultiply, KernelTypes); -TYPED_TEST(TestGemmMultiplyMultiply, Test_BF16FP16) { this->Run(); } +TYPED_TEST(TestGemmMultiplyMultiply, Test) { this->Run(); }