Skip to content

Commit

Permalink
Delete unused QAT utils code (#1579)
Browse files Browse the repository at this point in the history
  • Loading branch information
andrewor14 authored Jan 17, 2025
1 parent a1c67b9 commit 69f3795
Showing 1 changed file with 0 additions and 8 deletions.
8 changes: 0 additions & 8 deletions torchao/quantization/qat/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
"""
Expand Down

0 comments on commit 69f3795

Please sign in to comment.