Commit 84ed31d3 authored by Bartlomiej Kocot's avatar Bartlomiej Kocot
Browse files

Restore previous padd for 1x1 specialization

parent 04d762dc
...@@ -546,7 +546,7 @@ struct TransformConvBwdDataToGemm_v1 ...@@ -546,7 +546,7 @@ struct TransformConvBwdDataToGemm_v1
const auto wei_gemmbk0_gemmn_gemmbk1_grid_desc = const auto wei_gemmbk0_gemmn_gemmbk1_grid_desc =
ck::tensor_operation::device::PadTensorDescriptor( ck::tensor_operation::device::PadTensorDescriptor(
wei_gemmbk0_gemmnraw_gemmbk1_grid_desc, wei_gemmbk0_gemmnraw_gemmbk1_grid_desc,
make_tuple(GemmKPerBlock, GemmNPerBlock, BK1), make_tuple(BK0, GemmNPerBlock, BK1),
Sequence<false, DoPadGemmN, false>{}); Sequence<false, DoPadGemmN, false>{});
return wei_gemmbk0_gemmn_gemmbk1_grid_desc; return wei_gemmbk0_gemmn_gemmbk1_grid_desc;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment