File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed
backends/xnnpack/test/tester Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change 36
36
get_symmetric_quantization_config ,
37
37
XNNPACKQuantizer ,
38
38
)
39
- from torch .ao .quantization .quantizer .xnnpack_quantizer_utils import QuantizationConfig
39
+ from torch .ao .quantization .quantizer .xnnpack_quantizer_utils import (
40
+ convert_scalars_to_attrs ,
41
+ QuantizationConfig ,
42
+ )
40
43
from torch .testing import FileCheck
41
44
from torch .utils ._pytree import tree_flatten
42
45
@@ -105,6 +108,7 @@ def run(
105
108
self , artifact : torch .nn .Module , inputs : Optional [Tuple [torch .Tensor ]]
106
109
) -> None :
107
110
captured_graph = export .capture_pre_autograd_graph (artifact , inputs )
111
+ captured_graph = convert_scalars_to_attrs (captured_graph )
108
112
prepared = prepare_pt2e (captured_graph , self .quantizer )
109
113
converted = convert_pt2e (prepared )
110
114
self .converted_graph = converted
You can’t perform that action at this time.
0 commit comments