From 71d65d42944309b250f6c78de3cc9478ee5b1f95 Mon Sep 17 00:00:00 2001
From: apoorva <apoorva@streamhpc.com>
Date: Wed, 25 Jun 2025 08:24:32 +0000
Subject: [PATCH] Updated tests to ad BF16 instances as per review comment

---
 test/gemm_add/test_gemm_add_wmma.cpp | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/test/gemm_add/test_gemm_add_wmma.cpp b/test/gemm_add/test_gemm_add_wmma.cpp
index 5b6e9629a3..f4d29311b8 100644
--- a/test/gemm_add/test_gemm_add_wmma.cpp
+++ b/test/gemm_add/test_gemm_add_wmma.cpp
@@ -25,10 +25,9 @@ class TestGemmAdd : public TestGemmD0Common<Tuple>
     }
 };
 
-using KernelTypes = ::testing::Types<std::tuple<F16, F16, F32, F16, F16, Row, Row, Row, Row>,
-                                     std::tuple<F16, F16, F32, F16, F16, Row, Col, Row, Row>,
-                                     std::tuple<F16, F16, F32, F16, F16, Col, Row, Row, Row>,
-                                     std::tuple<F16, F16, F32, F16, F16, Col, Col, Row, Row>>;
+using KernelTypes =
+    ::testing::Types<std::tuple<F16, F16, F32, F16, F16, Row, Row, ck::Tuple<Row>, Row>,
+                     std::tuple<BF16, BF16, F32, BF16, BF16, Row, Row, ck::Tuple<Row>, Row>>;
 
 TYPED_TEST_SUITE(TestGemmAdd, KernelTypes);
 TYPED_TEST(TestGemmAdd, Test_BF16FP16) { this->Run(); }