gpu: jit: conv: handle scalar zero-points properly

oneapi-src · May 3, 2023 · cb91693 · cb91693
1 parent 85e58af
commit cb91693
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/gpu/jit/conv/zp_plan.cpp b/src/gpu/jit/conv/zp_plan.cpp
@@ -392,7 +392,9 @@ 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 = (kind_ == zp_comp_kind_t::wei_Xb_s16) ? 1 : 0;
+        int zp_stride = (kind_ == zp_comp_kind_t::wei_Xb_s16 && !is_zp_common())
+                ? 1
+                : 0;
         int wei_stride = 2;
         auto zp_type = zp_layout_.type();
         auto wei_type = wei_layout_.type();