Skip to content

Commit eb2a8fc

Browse files
authored
Unbreak BC, add back PerRow, PerTensor imports (#3396)
**Summary:** OSS users like sglang are still importing these from `torchao.quantization.observer`. Here we quickly unbreak BC since that was not the intention of #3370. **Test Plan:** CI
1 parent 170f696 commit eb2a8fc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

‎torchao/quantization/observer.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212

1313
from torchao.quantization.quant_primitives import _fake_quantize_affine
1414

15-
from .granularity import Granularity
15+
from .granularity import Granularity, PerRow, PerTensor # noqa: F401
1616
from .quant_primitives import (
1717
MappingType,
1818
ZeroPointDomain,

0 commit comments

Comments
 (0)