Fix slice quant pattern match bug

jackzhxng · web-flow · commit dd4fbd8c9f33 · 2025-11-05T18:06:29.000-05:00
diff --git a/exir/passes/_quant_patterns_and_replacements.py b/exir/passes/_quant_patterns_and_replacements.py
@@ -968,7 +968,7 @@ def pattern(x, dim, start, end, x_scale, x_zero_point, x_qmin, x_qmax):
             x, x_scale, x_zero_point, x_qmin, x_qmax, torch.uint8
         )
         x = torch.ops.aten.slice_copy.Tensor(x, dim, start, end)
-        x = torch.ops.quantized_decomposed.dequantize_per_tensor.default(
+        x = torch.ops.quantized_decomposed.quantize_per_tensor.default(
             x, x_scale, x_zero_point, x_qmin, x_qmax, torch.uint8
         )
         return x

Original file line number	Diff line number	Diff line change
`@@ -968,7 +968,7 @@ def pattern(x, dim, start, end, x_scale, x_zero_point, x_qmin, x_qmax):`
`968`	`968`	`x, x_scale, x_zero_point, x_qmin, x_qmax, torch.uint8`
`969`	`969`	`)`
`970`	`970`	`x = torch.ops.aten.slice_copy.Tensor(x, dim, start, end)`
`971`		`- x = torch.ops.quantized_decomposed.dequantize_per_tensor.default(`
	`971`	`+ x = torch.ops.quantized_decomposed.quantize_per_tensor.default(`
`972`	`972`	`x, x_scale, x_zero_point, x_qmin, x_qmax, torch.uint8`
`973`	`973`	`)`
`974`	`974`	`return x`