File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
aten/src/ATen/native/mkldnn/xpu/detail Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -307,7 +307,7 @@ void quantized_matmul(
307
307
args.insert ({DNNL_ARG_ATTR_SCALES | DNNL_ARG_SRC, m1_sc_m});
308
308
if (m1_need_zp) {
309
309
m1_zp_m = dnnl_memory_from_host_scalar (
310
- static_cast <int32_t >(input_zero_point), m1_zp_tensor, engine);
310
+ static_cast <int32_t >(input_zero_point), m1_zp_tensor, engine);
311
311
args.insert ({DNNL_ARG_ATTR_ZERO_POINTS | DNNL_ARG_SRC, m1_zp_m});
312
312
}
313
313
@@ -318,7 +318,7 @@ void quantized_matmul(
318
318
args.insert ({DNNL_ARG_ATTR_SCALES | DNNL_ARG_DST, dst_sc_m});
319
319
if (dst_need_zp) {
320
320
dst_zp_m = dnnl_memory_from_host_scalar (
321
- static_cast <int32_t >(output_zero_point), dst_zp_tensor, engine);
321
+ static_cast <int32_t >(output_zero_point), dst_zp_tensor, engine);
322
322
args.insert ({DNNL_ARG_ATTR_ZERO_POINTS | DNNL_ARG_DST, dst_zp_m});
323
323
}
324
324
You can’t perform that action at this time.
0 commit comments