Skip to content

Commit 8d1480b

Browse files
authored
Make the quantizer type a config argument
Differential Revision: D69704507 Pull Request resolved: #8510
1 parent f174ca8 commit 8d1480b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

backends/cadence/aot/quantizer/quantizer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66

77
# pyre-strict
88

9+
from dataclasses import dataclass
910
from typing import List, Optional, Tuple, Union
1011

1112
import torch
@@ -177,6 +178,8 @@ def get_cadence_default_quantizers() -> List[Quantizer]:
177178
]
178179

179180

181+
# Note: need dataclass to be used in CI configs through OmegaConf and Hydra
182+
@dataclass
180183
class CadenceQuantizer(ComposableQuantizer):
181184
"""
182185
Generic CadenceQuantizer. Although it can be used directly, it is typically a base

0 commit comments

Comments
 (0)