Skip to content

Commit 83f607e

Browse files
authored
[CK Tile] Fix building grouped conv examples in CK Tile (#2777)
* Fix compilation of the grouped conv examples. * Fix grouped conv bwd weight example output in CK Tile.
1 parent c6010f2 commit 83f607e

File tree

3 files changed

+5
-7
lines changed

3 files changed

+5
-7
lines changed

example/ck_tile/20_grouped_convolution/run_grouped_convolution_bwd_weight_example.inc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ int run_grouped_conv_bwd_weight_example_with_layouts(
130130
output_dev_buf.GetDeviceBuffer(),
131131
kbatch);
132132

133-
std::cout << "Run Grouped Conv Fwd kernel" << std::endl;
133+
std::cout << "Run Grouped Conv Bwd Weight kernel" << std::endl;
134134
std::cout << "input: " << input.mDesc << std::endl;
135135
std::cout << "weight: " << weight.mDesc << std::endl;
136136
std::cout << "output: " << output.mDesc << std::endl;

include/ck_tile/ops/grouped_convolution/kernel/grouped_convolution_backward_data_kernel.hpp

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -384,9 +384,8 @@ struct GroupedConvBwdDataKernelArgs
384384

385385
static constexpr index_t MaxGroupedGemmGroupsNum = 128;
386386

387-
using ABCGridDescs =
388-
remove_cvref_t<decltype(ConvToGemmTransformer{}
389-
.template MakeABCGridDescriptor_A_K0_M_K1_B_K0_N_K1_C_M_N(1))>;
387+
using ABCGridDescs = remove_cvref_t<
388+
decltype(ConvToGemmTransformer{}.MakeABCGridDescriptor_A_K0_M_K1_B_K0_N_K1_C_M_N(1))>;
390389

391390
using AGridDescMK = remove_cvref_t<decltype(ABCGridDescs{}[number<0>{}])>;
392391
using BGridDescNK = remove_cvref_t<decltype(ABCGridDescs{}[number<1>{}])>;

include/ck_tile/ops/grouped_convolution/kernel/grouped_convolution_backward_weight_kernel.hpp

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -254,9 +254,8 @@ struct GroupedConvBwdWeightKernelArgs
254254
GemmBatch = args.G_;
255255
}
256256

257-
using ABCGridDescs =
258-
remove_cvref_t<decltype(ConvToGemmTransformer{}
259-
.template MakeABCGridDescriptor_A_K0_M_K1_B_K0_N_K1_C_M_N())>;
257+
using ABCGridDescs = remove_cvref_t<
258+
decltype(ConvToGemmTransformer{}.MakeABCGridDescriptor_A_K0_M_K1_B_K0_N_K1_C_M_N())>;
260259

261260
using AGridDescMK = remove_cvref_t<decltype(ABCGridDescs{}[number<0>{}])>;
262261
using BGridDescNK = remove_cvref_t<decltype(ABCGridDescs{}[number<1>{}])>;

0 commit comments

Comments
 (0)