ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 06:49:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CK_BUILDER] Clean-up fwd conv builder implementation (#3110)

Browse Source
This commit is contained in:
Ville Pietilä
2025-10-29 05:37:33 +02:00
committed by GitHub
parent 515e283091
commit 13e13ce359
10 changed files with 84 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
experimental/builder/test/utils/ckb_conv_test_common.hpp
View File

@@ -10,11 +10,6 @@ namespace ck_tile::builder::test_utils {
using namespace ck_tile::builder;
using namespace test;
// Common test base class
class FwdConvBuilderTestBase : public ::testing::Test
{
};
// Common test implementation
template <auto FwdConvSignature,
ThreadBlock FwdThreadBlock,
@@ -93,11 +88,4 @@ constexpr void run_test()
EXPECT_NE(invoker_ptr, nullptr);
}
// Common thread block configurations
constexpr ThreadBlock DefaultThreadBlock{.block_size = 256,
.tile_size = {.m = 256, .n = 256, .k = 32}};
constexpr ThreadBlock SmallThreadBlock{.block_size = 256,
.tile_size = {.m = 128, .n = 128, .k = 32}};
} // namespace ck_tile::builder::test_utils