Skip to content

Commit

Permalink
Update on "Move and rename GranularityType -> Granularity"
Browse files Browse the repository at this point in the history
Summary: Move GranularityType to quant_primitives.py to be
consistent with other similar fields like MappingType and
ZeroPointDomain.

Test Plan: CI

[ghstack-poisoned]
  • Loading branch information
andrewor14 committed Oct 9, 2024
2 parents f7f7864 + fd6b477 commit 7ca6d56
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchao/prototype/awq/api.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
import torch
import torch.nn.functional as F

from torchao.quantization.granularity import PerGroup
from torchao.quantization.quant_primitives import (
MappingType,
PerGroup,
ZeroPointDomain,
_DTYPE_TO_QVALUE_BOUNDS,
)
Expand Down

0 comments on commit 7ca6d56

Please sign in to comment.