From b8dfe42cad9b02b3b6e2562f58f7c7fba7de434d Mon Sep 17 00:00:00 2001
From: "Chereshnev, Eugene" <eugene.chereshnev@intel.com>
Date: Wed, 17 May 2023 21:08:31 -0700
Subject: [PATCH] gpu: jit: reorder: fix op_plan_t::shift_offset()

---
 src/gpu/jit/codegen/reorder.hpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/gpu/jit/codegen/reorder.hpp b/src/gpu/jit/codegen/reorder.hpp
index 3b2e870ce32..8a5c3f84c29 100644
--- a/src/gpu/jit/codegen/reorder.hpp
+++ b/src/gpu/jit/codegen/reorder.hpp
@@ -61,8 +61,8 @@ struct op_plan_t {
 
         for (int i = 0; i < original_esize; i += esize) {
             fixup(op, mod, dst, src);
-            shift_offset(dst, esize);
-            shift_offset(src, esize);
+            shift_offset(dst, esize * dst.getHS());
+            shift_offset(src, esize * src.getHS());
         }
     }