Commit e30ae373 authored by rocking's avatar rocking
Browse files

Fix kernel parameter for NDHWC, fastest dimension C is not reduced

parent c26c154e
......@@ -85,7 +85,7 @@ bool pool3d_bwd_test(bool do_verification,
1, // ReduceMThreadSliceSize
1, // ReduceKThreadSliceSize
1, // InSrcOutDstVectorSize
true>;
false>;
auto OutSpatialLength = [&](auto InSpatialLength, int index) {
ck::index_t left_pad = dinput_left_pads[index];
......
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