Skip to content

Commit

Permalink
Add missing QAT linear classes in BC path
Browse files Browse the repository at this point in the history
Summary: #1037 moved the
module API and attempted to maintain BC, but it seems it
missed the linear classes. This unintentionally broke some
workflows. This commit adds the missing linear classes back
until we are ready to fully remove this path.
  • Loading branch information
andrewor14 committed Oct 16, 2024
1 parent 48bc81c commit 025375d
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions torchao/quantization/prototype/qat/_module_swap_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,9 @@
# These will be removed in the future

from .linear import (
Int8DynActInt4WeightQATLinear,
Int8DynActInt4WeightQATQuantizer as Int8DynActInt4WeightQATQuantizerModuleSwap,
Int4WeightOnlyQATLinear,
Int4WeightOnlyQATQuantizer as Int4WeightOnlyQATQuantizerModuleSwap,
enable_8da4w_fake_quant as enable_8da4w_fake_quant_module_swap,
disable_8da4w_fake_quant as disable_8da4w_fake_quant_module_swap,
Expand Down

0 comments on commit 025375d

Please sign in to comment.