gpu: jit: reorder: fix op_plan_t::shift_offset()

oneapi-src · May 19, 2023 · b8dfe42 · b8dfe42
1 parent 276cc21
commit b8dfe42
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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());
         }
     }