Skip to content

Commit 025375d

Browse files
committed
Add missing QAT linear classes in BC path
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.
1 parent 48bc81c commit 025375d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

torchao/quantization/prototype/qat/_module_swap_api.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,9 @@
22
# These will be removed in the future
33

44
from .linear import (
5+
Int8DynActInt4WeightQATLinear,
56
Int8DynActInt4WeightQATQuantizer as Int8DynActInt4WeightQATQuantizerModuleSwap,
7+
Int4WeightOnlyQATLinear,
68
Int4WeightOnlyQATQuantizer as Int4WeightOnlyQATQuantizerModuleSwap,
79
enable_8da4w_fake_quant as enable_8da4w_fake_quant_module_swap,
810
disable_8da4w_fake_quant as disable_8da4w_fake_quant_module_swap,

0 commit comments

Comments
 (0)