ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-05 14:11:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add the last two forward instance traits. (#3134)

Browse Source 
* Add InstanceTraits for DeviceGroupedConvFwdMultipleD_Wmma_CShuffle

* Add InstanceTraits for kernel_grouped_conv_fwd_dl_multiple_d

* A few small changes to fix broken instance traits.
This commit is contained in:
John Shumway
2025-10-31 07:52:42 -07:00
committed by GitHub
parent 1977e4b96a
commit 5ed2046bee
17 changed files with 1207 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
experimental/builder/test/test_instance_traits_util.cpp
View File

@@ -202,8 +202,8 @@ TEST(InstanceTraitsUtil, PipelineVersionNameReturnsCorrectStrings)
TEST(InstanceTraitsUtil, LoopSchedulerNameReturnsCorrectStrings)
{
using enum ck::LoopScheduler;
EXPECT_THAT(std::vector<std::string_view> names = {loop_scheduler_name(Default),
loop_scheduler_name(Interwave)},
EXPECT_THAT((std::vector<std::string_view>{loop_scheduler_name(Default),
loop_scheduler_name(Interwave)}),
ElementsAre("Default", "Interwave"));
}
@@ -267,5 +267,15 @@ TEST(InstanceTraitsUtil, SequenceNameReturnsSeqStringForMultipleValueSequence)
EXPECT_EQ((sequence_name<ck::Sequence<256, 128, 64, 32, 16>>()), "Seq(256,128,64,32,16)");
}
TEST(InstanceTraitsUtil, TypeOrTypeTupleNameReturnsCorrectStringForScalarDataType)
{
EXPECT_EQ(type_or_type_tuple_name<float>(), "fp32");
}
TEST(InstanceTraitsUtil, TypeOrTypeTupleNameReturnsCorrectStringForTupleOfDataTypes)
{
EXPECT_EQ((type_or_type_tuple_name<ck::Tuple<ck::half_t, float>>()), "Tuple(fp16,fp32)");
}
} // namespace
} // namespace ck_tile::reflect::detail