From 9831c30427ef477b3cf3891b9cb628970b6fa7d8 Mon Sep 17 00:00:00 2001 From: Scott Roy <161522778+metascroy@users.noreply.github.com> Date: Wed, 11 Feb 2026 11:51:50 -0800 Subject: [PATCH] Revert "Fix KeyError in InsertIOQDQ pass for LLM quantization (#17194)" This reverts commit 08282a4cc7a0ceefea40fba63feb7f8f55a61f22. --- backends/qualcomm/_passes/insert_io_qdq.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/backends/qualcomm/_passes/insert_io_qdq.py b/backends/qualcomm/_passes/insert_io_qdq.py index f239aac2519..caecae64fa8 100644 --- a/backends/qualcomm/_passes/insert_io_qdq.py +++ b/backends/qualcomm/_passes/insert_io_qdq.py @@ -31,16 +31,11 @@ class InsertIOQDQ(ExportPass): """ q_dq_map = { - # per tensor (quantize -> dequantize) + # per tensor exir_ops.edge.quantized_decomposed.quantize_per_tensor.default: exir_ops.edge.quantized_decomposed.dequantize_per_tensor.tensor, exir_ops.edge.quantized_decomposed.quantize_per_tensor.tensor: exir_ops.edge.quantized_decomposed.dequantize_per_tensor.tensor, - # per tensor (dequantize -> dequantize, for nodes with dequantize encoding) - exir_ops.edge.quantized_decomposed.dequantize_per_tensor.default: exir_ops.edge.quantized_decomposed.dequantize_per_tensor.default, - exir_ops.edge.quantized_decomposed.dequantize_per_tensor.tensor: exir_ops.edge.quantized_decomposed.dequantize_per_tensor.tensor, - # per channel (quantize -> dequantize) + # per channel exir_ops.edge.quantized_decomposed.quantize_per_channel.default: exir_ops.edge.quantized_decomposed.dequantize_per_channel.default, - # per channel (dequantize -> dequantize, for nodes with dequantize encoding) - exir_ops.edge.quantized_decomposed.dequantize_per_channel.default: exir_ops.edge.quantized_decomposed.dequantize_per_channel.default, } def __init__(self, edge_program: torch.export.ExportedProgram):