4c3f59b70e98acf5bca2c1c26ebee5a018be0de4,torch/quantization/fx/quantization_patterns.py,Add,convert,#Add#Any#Any#Any#Any#Any#,92

Before Change

            else:
                op = torch.ops.quantized.add
            kwargs = {**self.add_node.kwargs, "scale": scale, "zero_point": zero_point}
            return quantizer.quantized_graph.create_node(
                "call_function", op, load_arg(quantized=True)(self.add_node.args), kwargs)

// TODO: merge with Add
@register_quant_pattern(operator.mul)
@register_quant_pattern(torch.mul)

After Change

                op = torch.ops.quantized.add
            kwargs = {**self.add_node.kwargs}
            add_args = (*load_arg(quantized=True)(self.add_node.args), scale_arg, zero_point_arg)
            op = quantizer.quantized_graph.create_node(
                "call_function", op, add_args, kwargs)
            return op

// TODO: merge with Add
@register_quant_pattern(operator.mul)
@register_quant_pattern(torch.mul)

Instances