Skip to content

Commit 85e58af

Browse files
committed
gpu: jit: conv: fix invalid access with zero-points
1 parent 97ac885 commit 85e58af

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/gpu/jit/conv/zp_plan.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -392,7 +392,7 @@ class zp_comp_init_plan_t : public base_plan_t {
392392

393393
stmt_t create_tile_wei_Xy_s16(
394394
const expr_t &zp, const expr_t &wei, const expr_t &comp) const {
395-
int zp_stride = is_zp_common() ? 0 : 1;
395+
int zp_stride = (kind_ == zp_comp_kind_t::wei_Xb_s16) ? 1 : 0;
396396
int wei_stride = 2;
397397
auto zp_type = zp_layout_.type();
398398
auto wei_type = wei_layout_.type();

0 commit comments

Comments
 (0)