diff --git a/test/gemm_add/test_gemm_add_add_fastgelu_wmma.cpp b/test/gemm_add/test_gemm_add_add_fastgelu_wmma.cpp index 2cde4c7ea3..a7d7e76395 100644 --- a/test/gemm_add/test_gemm_add_add_fastgelu_wmma.cpp +++ b/test/gemm_add/test_gemm_add_add_fastgelu_wmma.cpp @@ -35,4 +35,4 @@ using KernelTypes = ::testing::Types>; TYPED_TEST_SUITE(TestGemmAddAddFastgelu, KernelTypes); -TYPED_TEST(TestGemmAddAddFastgelu, Test_BF16FP16) { this->Run(); } +TYPED_TEST(TestGemmAddAddFastgelu, Test_FP16FP16) { this->Run(); } diff --git a/test/gemm_add/test_gemm_bilinear_wmma.cpp b/test/gemm_add/test_gemm_bilinear_wmma.cpp index 70a2b194c1..00888a12ca 100644 --- a/test/gemm_add/test_gemm_bilinear_wmma.cpp +++ b/test/gemm_add/test_gemm_bilinear_wmma.cpp @@ -66,4 +66,4 @@ using KernelTypes = ::testing::Types>; TYPED_TEST_SUITE(TestGemmBilinear, KernelTypes); -TYPED_TEST(TestGemmBilinear, Test) { this->Run({{16, 32, 64}, {512, 2048, 4096}, {2048, 1024, 16}}); } +TYPED_TEST(TestGemmBilinear, Test) { this->Run(); }