From 1a1d411513cf15e48a0d88a45568d7a4b688e796 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tadej=20Ciglari=C4=8D?= <tadej.ciglaric@codeplay.com>
Date: Fri, 20 Sep 2024 13:19:06 +0200
Subject: [PATCH] generic: sycl: bugfix post ops with blocked formats

---
 src/gpu/generic/sycl/binary_kernels.hpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/gpu/generic/sycl/binary_kernels.hpp b/src/gpu/generic/sycl/binary_kernels.hpp
index 7e89e73bc16..4c05939e42b 100644
--- a/src/gpu/generic/sycl/binary_kernels.hpp
+++ b/src/gpu/generic/sycl/binary_kernels.hpp
@@ -129,8 +129,10 @@ struct binary_kernel_vec_t {
 
                 auto acc = compute_alg_n(src0, src1, conf_.alg_kind);
 
-                acc = conf_.post_ops.apply(acc, dst_, idx, po_args_, off_dst);
-                dst_mem.store_md(acc, off_dst);
+                int dst_idx = dst_mem.md().off_v(off_dst);
+                acc = conf_.post_ops.apply(
+                        acc, dst_, dst_idx, po_args_, off_dst);
+                dst_mem.store(acc, dst_idx);
             }
         }
     }