Skip to content

Commit b7f32ae

Browse files
committed
Update on "Remove old QAT APIs"
**Summary:** As a follow-up to #2641, which deprecated the old QAT APIs in 0.13.0, we remove them now in the next release 0.15.0. Fixes #2630. **Test Plan:** CI [ghstack-poisoned]
2 parents 9483ff0 + d624214 commit b7f32ae

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

torchao/quantization/prototype/qat/api.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
from torchao.quantization.qat.api import (
22
ComposableQATQuantizer,
3+
)
4+
from torchao.quantization.qat.api import (
35
IntxFakeQuantizeConfig as FakeQuantizeConfig,
46
)
57

torchao/quantization/quant_api.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,6 @@
148148
"int8_dynamic_activation_int8_semi_sparse_weight",
149149
"int4_weight_only",
150150
"int8_weight_only",
151-
"intx_quantization_aware_training",
152151
"float8_weight_only",
153152
"uintx_weight_only",
154153
"fpx_weight_only",

torchao/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from functools import reduce
1212
from importlib.metadata import version
1313
from math import gcd
14-
from typing import Any, Callable, Optional, Type
14+
from typing import Any, Callable, Optional
1515

1616
import torch
1717
import torch.nn.utils.parametrize as parametrize

0 commit comments

Comments
 (0)