From 0696f999375c966c748554c5daa8efd0bfeac366 Mon Sep 17 00:00:00 2001
From: apoorva <apoorva@streamhpc.com>
Date: Fri, 13 Jun 2025 08:30:41 +0000
Subject: [PATCH] Added tests.

---
 test/gemm_add/CMakeLists.txt         |  5 ++++
 test/gemm_add/test_gemm_add_wmma.cpp | 40 ++++++++++++++++++++++++++++
 2 files changed, 45 insertions(+)
 create mode 100644 test/gemm_add/test_gemm_add_wmma.cpp

diff --git a/test/gemm_add/CMakeLists.txt b/test/gemm_add/CMakeLists.txt
index f7430b8ae1..8f099f1286 100644
--- a/test/gemm_add/CMakeLists.txt
+++ b/test/gemm_add/CMakeLists.txt
@@ -22,3 +22,8 @@ add_gtest_executable(test_gemm_add_fastgelu_wmma test_gemm_add_fastgelu_wmma.cpp
 if(result EQUAL 0)
     target_link_libraries(test_gemm_add_fastgelu_wmma PRIVATE utility device_gemm_add_fastgelu_instance)
 endif()
+
+add_gtest_executable(test_gemm_add_wmma test_gemm_add_wmma.cpp)
+if(result EQUAL 0)
+    target_link_libraries(test_gemm_add_wmma PRIVATE utility device_gemm_add_instance)
+endif()
diff --git a/test/gemm_add/test_gemm_add_wmma.cpp b/test/gemm_add/test_gemm_add_wmma.cpp
new file mode 100644
index 0000000000..a2238463f5
--- /dev/null
+++ b/test/gemm_add/test_gemm_add_wmma.cpp
@@ -0,0 +1,40 @@
+// SPDX-License-Identifier: MIT
+// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
+
+#include "gtest/gtest.h"
+#include "ck/ck.hpp"
+#include "profiler/profile_gemm_add_impl.hpp"
+#include "test_gemm_common.hpp"
+
+template <typename Tuple>
+class TestGemmAdd : public TestGemmD0Common<Tuple>
+{
+    private:
+    using ADataType   = std::tuple_element_t<0, Tuple>;
+    using BDataType   = std::tuple_element_t<1, Tuple>;
+    using AccDataType = std::tuple_element_t<2, Tuple>;
+    using D0DataType  = std::tuple_element_t<3, Tuple>;
+    using EDataType   = std::tuple_element_t<4, Tuple>;
+    using ALayout     = std::tuple_element_t<5, Tuple>;
+    using BLayout     = std::tuple_element_t<6, Tuple>;
+    using D0Layout    = std::tuple_element_t<7, Tuple>;
+    using ELayout     = std::tuple_element_t<8, Tuple>;
+
+    constexpr static auto ProfileGemmAddImpl = ck::profiler::profile_gemm_add_impl<ADataType,
+                                                                                   BDataType,
+                                                                                   AccDataType,
+                                                                                   D0DataType,
+                                                                                   EDataType,
+                                                                                   ALayout,
+                                                                                   BLayout,
+                                                                                   D0Layout,
+                                                                                   ELayout>;
+
+    decltype(ProfileGemmAddImpl) GetImpl() override { return ProfileGemmAddImpl; }
+};
+
+using KernelTypes = ::testing::Types<std::tuple<F16, , F16, F32, F16, F16, Row, Row, Row, Row>,
+                                     std::tuple<BF16, BF16, F32, BF16, BF16, Row, Row, Row, Row>>;
+
+TYPED_TEST_SUITE(TestGemmAdd, KernelTypes);
+TYPED_TEST(TestGemmAdd, Test_BF16FP16) { this->Run(); }