fix if branch

pytorch · Oct 22, 2024 · 3000e05 · 3000e05
1 parent c9a66a7
commit 3000e05
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchao/quantization/quant_api.py b/torchao/quantization/quant_api.py
@@ -614,9 +614,9 @@ def get_weight_block_size(x):
 
         # input settings
         if act_mapping_type == MappingType.SYMMETRIC:
-            input_quant_func = _int8_asymm_per_token_quant
-        else:
             input_quant_func = _int8_symm_per_token_reduced_range_quant
+        else:
+            input_quant_func = _int8_asymm_per_token_quant
 
         block_size = get_weight_block_size(weight)
         weight = to_affine_quantized_intx(weight, mapping_type, block_size, target_dtype, eps=eps, zero_point_dtype=zero_point_dtype, _layout=layout)