gpu: jit: conv: fix invalid access with zero-points

oneapi-src · Apr 27, 2023 · 85e58af · 85e58af
1 parent 97ac885
commit 85e58af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gpu/jit/conv/zp_plan.cpp b/src/gpu/jit/conv/zp_plan.cpp
@@ -392,7 +392,7 @@ class zp_comp_init_plan_t : public base_plan_t {
 
     stmt_t create_tile_wei_Xy_s16(
             const expr_t &zp, const expr_t &wei, const expr_t &comp) const {
-        int zp_stride = is_zp_common() ? 0 : 1;
+        int zp_stride = (kind_ == zp_comp_kind_t::wei_Xb_s16) ? 1 : 0;
         int wei_stride = 2;
         auto zp_type = zp_layout_.type();
         auto wei_type = wei_layout_.type();