File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
tests/test_quantization/lifecycle Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -250,7 +250,7 @@ def calibrate_input_hook(_, args):
250
250
global_scale = linear .input_observer .get_global_scale (args [0 ])
251
251
linear .input_global_scale .data = global_scale
252
252
253
- if linear .quantization_scheme .input_activations .dynamic == False :
253
+ if linear .quantization_scheme .input_activations .dynamic is False :
254
254
scale , zero_point = linear .input_observer (args [0 ])
255
255
linear .input_scale .data = scale
256
256
linear .input_zero_point .data = zero_point
@@ -331,7 +331,7 @@ def test_static_attention_quantization(
331
331
attention .k_observer = MockMinMaxObserver (attention , base_name = "k" )
332
332
333
333
# calibrate quantization parameters
334
- if scheme .input_activations .dynamic == False :
334
+ if scheme .input_activations .dynamic is False :
335
335
scale , zero_point = attention .k_observer (input )
336
336
attention .k_scale .data = scale
337
337
attention .k_zero_point .data = zero_point
You can’t perform that action at this time.
0 commit comments