Skip to content

Commit 69c90fc

Browse files
Dylan Angusdylanangus
Dylan Angus
authored andcommittedMar 13, 2024
gpu: miopen: pooling: Fixed strides in 1D pooling
1 parent 6efe933 commit 69c90fc

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed
 

‎src/gpu/amd/miopen_pooling_impl.hpp

+4
Original file line numberDiff line numberDiff line change
@@ -86,9 +86,13 @@ struct miopen_pooling_impl_base_t {
8686
// [n, c, w, 1]
8787
dims_[src][3] = dims_[src][2];
8888
dims_[src][2] = 1;
89+
strides_[src][2] = dims_[src][3];
90+
strides_[src][3] = 1;
8991

9092
dims_[dst][3] = dims_[dst][2];
9193
dims_[dst][2] = 1;
94+
strides_[dst][2] = dims_[dst][3];
95+
strides_[dst][3] = 1;
9296

9397
// Set kernel dimensions to [1, kw]
9498
kernel_dims_[1] = kernel_dims_[0];

0 commit comments

Comments
 (0)