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

fixes

Browse Source
This commit is contained in:
Bartlomiej Kocot
2025-06-13 13:36:42 +00:00
parent 0a3b31a6f3
commit 1708cbc6e6
5 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
example/ck_tile/37_grouped_convolution/CMakeLists.txt → example/ck_tile/19_grouped_convolution/CMakeLists.txt
View File

4
example/ck_tile/37_grouped_convolution/grouped_convolution_forward.cpp → example/ck_tile/19_grouped_convolution/grouped_convolution_forward.cpp
View File

@@ -20,12 +20,10 @@ template <ck_tile::index_t NDimSpatial,
typename InLayout,
typename WeiLayout,
typename OutLayout>
float grouped_conv_fwd_calc(const ck_tile::GroupedConvHostArgs& args,
const ck_tile::stream_config& s)
float grouped_conv_fwd(const ck_tile::GroupedConvHostArgs& args, const ck_tile::stream_config& s)
{
constexpr int kBlockPerCu = 1;
// This part comes from the Codegen
constexpr ck_tile::index_t M_Tile = 64;
constexpr ck_tile::index_t N_Tile = 64;
constexpr ck_tile::index_t K_Tile = 64;

7
example/ck_tile/37_grouped_convolution/grouped_convolution_utils.hpp → example/ck_tile/19_grouped_convolution/grouped_convolution_utils.hpp
View File

@@ -23,7 +23,10 @@ ck_tile::index_t fill_spatial_dimensions(std::vector<ck_tile::index_t>& filter_s
constexpr ck_tile::index_t non_sp_dims = 3;
const ck_tile::index_t n_dim_sp = arg_parser.get_str("in_layout").size() - non_sp_dims;
assert(n_dim_sp >= 1 && n_dim_sp <= 3 && "Wrong layout!");
if(!(n_dim_sp >= 1 && n_dim_sp <= 3))
{
throw std::runtime_error("Wrong layout!\n");
}
if(n_dim_sp == 3)
{
@@ -101,4 +104,4 @@ auto create_args(int argc, char* argv[])
}
// host API
float gemm_calc(const ck_tile::GroupedConvHostArgs& args, const ck_tile::stream_config& s);
float grouped_conv_fwd(const ck_tile::GroupedConvHostArgs& args, const ck_tile::stream_config& s);

4
example/ck_tile/37_grouped_convolution/run_grouped_convolution_example.inc → example/ck_tile/19_grouped_convolution/run_grouped_convolution_example.inc
View File

@@ -116,8 +116,8 @@ int run_grouped_conv_fwd_example_with_layouts(
if(init_method == 0)
{
ck_tile::FillUniformDistribution<InDataType>{-1.f, 1.f}(input);
ck_tile::FillUniformDistribution<WeiDataType>{-1.f, 1.f}(weight);
ck_tile::FillUniformDistribution<InDataType>{-5.f, 5.f}(input);
ck_tile::FillUniformDistribution<WeiDataType>{-5.f, 5.f}(weight);
}
else if(init_method == 1)
{

2
example/ck_tile/CMakeLists.txt
View File

@@ -18,6 +18,6 @@ add_subdirectory(15_fused_moe)
add_subdirectory(16_batched_gemm)
add_subdirectory(17_grouped_gemm)
add_subdirectory(18_flatmm)
add_subdirectory(19_grouped_convolution)
add_subdirectory(35_batched_transpose)
add_subdirectory(36_copy)
add_subdirectory(37_grouped_convolution)