Skip to content

Commit 8a5153b

Browse files
authored
Share qparams for aten.concat.default in composable (V2) quantizer (pytorch#20999)
Differential Revision: D112236028 Pull Request resolved: pytorch#20999
1 parent 32f56be commit 8a5153b

2 files changed

Lines changed: 7 additions & 0 deletions

File tree

‎backends/arm/quantizer/arm_quantizer_utils.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -461,6 +461,7 @@ class SharedQspecQuantizer(Quantizer, QuantizerReporterUser):
461461
torch.ops.aten.pixel_shuffle.default,
462462
torch.ops.aten.pixel_unshuffle.default,
463463
torch.ops.aten.cat.default,
464+
torch.ops.aten.concat.default,
464465
torch.ops.aten.concatenate.default,
465466
torch.ops.aten.stack.default,
466467
torch.ops.aten.dropout.default,

‎backends/cortex_m/test/misc/test_portable_int8.py‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -365,6 +365,12 @@ def _quantize_and_export(
365365
(torch.randn(2, 3, 4, 5), torch.randn(2, 3, 4, 5)),
366366
None,
367367
),
368+
"concat": OpCase(
369+
torch.ops.aten.concat.default,
370+
_build_module(lambda x, y: torch.ops.aten.concat.default([x, y], 1)),
371+
(torch.randn(2, 3, 4, 5), torch.randn(2, 3, 4, 5)),
372+
None,
373+
),
368374
"concatenate": OpCase(
369375
torch.ops.aten.concatenate.default,
370376
_build_module(lambda x, y: torch.ops.aten.concatenate.default([x, y], 1)),

0 commit comments

Comments
 (0)