ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-17 00:58:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

implemented tests for instances from grouped_convolution_forward_convscale.hpp:210

Browse Source
This commit is contained in:
Kevin Abraham
2025-10-24 20:11:40 +00:00
parent bd40b58266
commit 02b250f298
1 changed files with 26 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
experimental/builder/test/test_ck_factory_grouped_convolution_forward_convscale.cpp
View File

@@ -19,6 +19,7 @@ using WeiLayout = ck::tensor_layout::convolution::GKZYXC;
using OutLayout = ck::tensor_layout::convolution::NDHWGK;
using ck::tensor_operation::device::DeviceGroupedConvFwdMultipleABD;
using ck::tensor_operation::device::instance::CombConvScale;
using ck::tensor_operation::device::instance::CombConvScaleRelu;
using ck::tensor_operation::element_wise::ConvInvscale;
using ck::tensor_operation::element_wise::ConvScale;
@@ -80,7 +81,7 @@ struct F8_ConvScale
};
};
struct F8_comb1_ConvScale
struct F8_BF8_comb1_ConvScale
{
using DeviceOp = ::DeviceOp<ck::Tuple<>,
ck::Tuple<>,
@@ -98,7 +99,7 @@ struct F8_comb1_ConvScale
};
};
struct F8_comb2_ConvScale
struct F8_BF8_comb2_ConvScale
{
using DeviceOp = ::DeviceOp<ck::Tuple<>,
ck::Tuple<>,
@@ -116,7 +117,7 @@ struct F8_comb2_ConvScale
};
};
struct F8_comb3_ConvScale
struct F8_BF8_comb3_ConvScale
{
using DeviceOp = ::DeviceOp<ck::Tuple<>,
ck::Tuple<>,
@@ -134,6 +135,24 @@ struct F8_comb3_ConvScale
};
};
struct F8_float_CombConvScale
{
using DeviceOp = ::DeviceOp<ck::Tuple<>,
ck::Tuple<>,
ck::f8_t,
ck::f8_t,
float,
CombConvScale,
ck::f8_t,
ck::f8_t>;
constexpr static auto expected = {
// clang-format off
""
// clang-format on
};
};
struct F8_ConvScaleRelu
{
using DeviceOp = ::DeviceOp<ck::Tuple<>,
@@ -201,9 +220,10 @@ struct F8_ConvInvscale
using TestTypes =
::testing::Types<F8_ConvScale,
F8_comb1_ConvScale,
F8_comb2_ConvScale,
F8_comb3_ConvScale,
F8_BF8_comb1_ConvScale,
F8_BF8_comb2_ConvScale,
F8_BF8_comb3_ConvScale,
F8_float_CombConvScale,
F8_ConvScaleRelu,
F8_CombConvScaleRelu,
F8_ConvScaleAdd, F8_ConvInvscale>;