File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed
tests/llmcompressor/observers Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -53,7 +53,7 @@ def test_min_max_observer_symmetric_scale_range():
53
53
tensor *= 127
54
54
55
55
num_bits = 8
56
- weights = QuantizationArgs (num_bits = num_bits , symmetric = True )
56
+ weights = QuantizationArgs (num_bits = num_bits , symmetric = True , observer = "minmax" )
57
57
58
58
observer = weights .observer
59
59
observer = Observer .load_from_registry (observer , quantization_args = weights )
@@ -82,7 +82,7 @@ def test_min_max_observer_value_update():
82
82
83
83
tensor = inp
84
84
num_bits = 8
85
- weights = QuantizationArgs (num_bits = num_bits , symmetric = True )
85
+ weights = QuantizationArgs (num_bits = num_bits , symmetric = True , observer = "minmax" )
86
86
observer = weights .observer
87
87
observer = Observer .load_from_registry (observer , quantization_args = weights )
88
88
curr_max = 1
@@ -107,7 +107,7 @@ def test_g_idx():
107
107
group_size = 2
108
108
input_shape = (128 , 512 )
109
109
tensor = torch .rand (input_shape )
110
- weights = QuantizationArgs (num_bits = 8 , group_size = group_size )
110
+ weights = QuantizationArgs (num_bits = 8 , group_size = group_size , observer = "minmax" )
111
111
g_idx = make_dummy_g_idx (tensor .shape [1 ], group_size )
112
112
113
113
observer = weights .observer
Original file line number Diff line number Diff line change @@ -46,7 +46,7 @@ def test_mse_observer_symmetric_scale_range():
46
46
tensor *= 127
47
47
48
48
num_bits = 8
49
- weights = QuantizationArgs (num_bits = num_bits , symmetric = True )
49
+ weights = QuantizationArgs (num_bits = num_bits , symmetric = True , observer = "mse" )
50
50
51
51
observer = weights .observer
52
52
observer = Observer .load_from_registry (observer , quantization_args = weights )
You can’t perform that action at this time.
0 commit comments