From bb7b307a40c9a3f6a0bb56b4bf2d0e40382b6ba5 Mon Sep 17 00:00:00 2001
From: Zoltan Lakatos <zoltan.lakatos@streamhpc.com>
Date: Mon, 30 Jun 2025 15:50:26 +0000
Subject: [PATCH] added missing changes

---
 test/gemm_add/test_gemm_add_add_fastgelu_wmma.cpp | 2 +-
 test/gemm_add/test_gemm_bilinear_wmma.cpp         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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<std::tuple<F16, F16, F32, F16, F16, F16, Ro
                                      std::tuple<F16, F16, F32, F16, F16, F16, Col, Col, Row, Row, Row>>;
 
 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<std::tuple<F16, F16, F32, F16, F16, Row, Ro
                                      std::tuple<I8, I8, I32, I8, I8, Col, Col, Row, Row>>;
 
 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(); }