ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-13 02:27:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add test for ak1 and bk1.

Browse Source 
These are not easy to access in the instance object, so just test the factory values.
This commit is contained in:
John Shumway
2025-09-02 21:06:16 +00:00
parent 834f0436a3
commit 97660c64e5
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
experimental/builder/include/ck_tile/builder/conv_builder.hpp
View File

@@ -29,9 +29,9 @@ struct ConvBuilder
{
using Signature = TSignature;
static constexpr auto kVersion = Version;
using builder = GroupedConvForwardXldCShuffleFactoryV3<Signature, ALGORITHM, Version>;
using factory = GroupedConvForwardXldCShuffleFactoryV3<Signature, ALGORITHM, Version>;
// Output: The kernel class.
using Instance = builder::Instance;
using Instance = factory::Instance;
};
} // namespace ck_tile::builder

4
experimental/builder/test/test_conv_builder.cpp
View File

@@ -51,6 +51,8 @@ TEST(ConvBuilderTest, TestConvFwdXdlBf16CompInstances2xInstance0)
Builder::Instance::TypeString(),
"DeviceGroupedConvFwdMultipleABD_Xdl_CShuffle_V3<256, 256, 128, 64, Default, 32, 32, 2, 2, "
"8, 8, 8, 1, 1, BlkGemmPipelineScheduler: Intrawave, BlkGemmPipelineVersion: v4>");
EXPECT_EQ(Builder::factory::TUNING.ak1, 16);
EXPECT_EQ(Builder::factory::TUNING.bk1, 16);
}
TEST(ConvBuilderTest, GroupedConvFwdXdlBf16CompInstance0)
{
@@ -62,6 +64,8 @@ TEST(ConvBuilderTest, GroupedConvFwdXdlBf16CompInstance0)
Builder::Instance::TypeString(),
"DeviceGroupedConvFwdMultipleABD_Xdl_CShuffle_V3<256, 256, 256, 32, Default, 32, 32, 4, 4, "
"8, 8, 8, 1, 1, BlkGemmPipelineScheduler: Intrawave, BlkGemmPipelineVersion: v4>");
EXPECT_EQ(Builder::factory::TUNING.ak1, 8);
EXPECT_EQ(Builder::factory::TUNING.bk1, 8);
};
} // namespace