From bdfdb0c11ec93c819d734afcc52945d8d43ffcd6 Mon Sep 17 00:00:00 2001 From: apoorva Date: Tue, 8 Jul 2025 12:17:20 +0000 Subject: [PATCH] Fixes applied according to review comments --- example/69_gemm_add_relu/CMakeLists.txt | 12 +++++------- test/gemm_add/test_gemm_add_relu_wmma.cpp | 5 ++--- 2 files changed, 7 insertions(+), 10 deletions(-) diff --git a/example/69_gemm_add_relu/CMakeLists.txt b/example/69_gemm_add_relu/CMakeLists.txt index fe9d783755..9ab3ef5a45 100644 --- a/example/69_gemm_add_relu/CMakeLists.txt +++ b/example/69_gemm_add_relu/CMakeLists.txt @@ -1,17 +1,15 @@ add_custom_target(example_gemm_add_relu_xdl) add_example_executable(example_gemm_add_relu_xdl_fp16 gemm_add_relu_xdl_fp16.cpp) +add_example_dependencies(example_gemm_add_relu_xdl example_gemm_add_relu_xdl_fp16) add_example_executable(example_gemm_add_relu_xdl_bf16 gemm_add_relu_xdl_bf16.cpp) - +add_example_dependencies(example_gemm_add_relu_xdl example_gemm_add_relu_xdl_bf16) add_custom_target(example_gemm_add_relu_wmma) + add_example_executable(example_gemm_add_relu_wmma_bf16 gemm_add_relu_wmma_bf16.cpp) +add_example_dependencies(example_gemm_add_relu_wmma example_gemm_add_relu_wmma_bf16) add_example_executable(example_gemm_add_relu_wmma_fp16 gemm_add_relu_wmma_fp16.cpp) - - - - - - +add_example_dependencies(example_gemm_add_relu_wmma example_gemm_add_relu_wmma_fp16) diff --git a/test/gemm_add/test_gemm_add_relu_wmma.cpp b/test/gemm_add/test_gemm_add_relu_wmma.cpp index e1e304f70f..8ff2f4217b 100644 --- a/test/gemm_add/test_gemm_add_relu_wmma.cpp +++ b/test/gemm_add/test_gemm_add_relu_wmma.cpp @@ -26,9 +26,8 @@ class TestGemmAddRelu : public TestGemmD0Common } }; -using KernelTypes = - ::testing::Types, Row>, - std::tuple, Row>>; +using KernelTypes = ::testing::Types, + std::tuple>; TYPED_TEST_SUITE(TestGemmAddRelu, KernelTypes); TYPED_TEST(TestGemmAddRelu, Test_BF16FP16) { this->Run(); }