From 69f3795a7b60bdc6b042b6c996f8c174fcd850c6 Mon Sep 17 00:00:00 2001 From: andrewor14 Date: Fri, 17 Jan 2025 14:03:11 -0500 Subject: [PATCH] Delete unused QAT utils code (#1579) --- torchao/quantization/qat/utils.py | 8 -------- 1 file changed, 8 deletions(-) diff --git a/torchao/quantization/qat/utils.py b/torchao/quantization/qat/utils.py index c901d59e92..80e909f48a 100644 --- a/torchao/quantization/qat/utils.py +++ b/torchao/quantization/qat/utils.py @@ -16,14 +16,6 @@ _get_per_token_block_size, ) -# Attribute name representing the forward prehook wrapping the -# linear input in an `AffineFakeQuantizedTensor` on a linear module. -# -# The value of this attribute is a 2-tuple of (prehook, handle). -# The prehook can be disabled by calling `handle.remove()`, and -# re-enabled by calling `module.register_forward_pre_hook(prehook)`. -_QAT_LINEAR_SUBCLASS_INPUT_PREHOOK = "_qat_linear_subclass_input_prehook" - class _GenericFakeQuantize(torch.autograd.Function): """