From 13bdeb9cb024247d803daffe5963c0943c6f0f1d Mon Sep 17 00:00:00 2001
From: Dylan Angus <dylan.angus@codeplay.com>
Date: Tue, 27 Feb 2024 19:23:16 +0000
Subject: [PATCH] gpu: miopen: pooling: Fixed strides in 1D pooling

---
 src/gpu/amd/miopen_pooling_impl.hpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/gpu/amd/miopen_pooling_impl.hpp b/src/gpu/amd/miopen_pooling_impl.hpp
index 01e623a7afe..f88cbf59372 100644
--- a/src/gpu/amd/miopen_pooling_impl.hpp
+++ b/src/gpu/amd/miopen_pooling_impl.hpp
@@ -86,9 +86,13 @@ struct miopen_pooling_impl_base_t {
             // [n, c, w, 1]
             dims_[src][3] = dims_[src][2];
             dims_[src][2] = 1;
+            strides_[src][2] = dims_[src][3];
+            strides_[src][3] = 1;
 
             dims_[dst][3] = dims_[dst][2];
             dims_[dst][2] = 1;
+            strides_[dst][2] = dims_[dst][3];
+            strides_[dst][3] = 1;
 
             // Set kernel dimensions to [1, kw]
             kernel_dims_[1] = kernel_dims_[0];