Fix bug in example

487d3751 · rocking · 7beb386f · 487d3751
Commit 487d3751 authored Jul 20, 2023 by rocking
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

example/51_avgpool3d_bwd/avgpool3d_bwd_common.hpp example/51_avgpool3d_bwd/avgpool3d_bwd_common.hpp +1 -1

No files found.
--- a/example/51_avgpool3d_bwd/avgpool3d_bwd_common.hpp
+++ b/example/51_avgpool3d_bwd/avgpool3d_bwd_common.hpp
@@ -77,7 +77,7 @@ bool pool3d_bwd_test(bool do_verification,

    ck::index_t Do = OutSpatialLength(Di, 0);
    ck::index_t Ho = OutSpatialLength(Hi, 1);
-    ck::index_t Wo = OutSpatialLength(Wi, 1);
+    ck::index_t Wo = OutSpatialLength(Wi, 2);

    Tensor<DOutDataType> dout(f_host_tensor_descriptor(N, C, Do, Ho, Wo, DOutLayout{}));
    Tensor<DInDataType> din_dev(f_host_tensor_descriptor(N, C, Di, Hi, Wi, DInLayout{}));