Fix typo

liangan1 · liangan1 · commit fa6ca5dd29f8 · 2025-05-22T18:59:32.000Z
diff --git a/torchao/dtypes/affine_quantized_tensor_ops.py b/torchao/dtypes/affine_quantized_tensor_ops.py
@@ -235,8 +235,8 @@ def _register_aqt_quantized_linear_dispatches():
             _linear_q_dq_impl,
         ),
         (
-            _linear_bf16_act_uint4_weight_int8_zero_check,
-            _linear_bf16_act_uint4_weight_int8_zero_impl,
+            _linear_fp_act_uint4_weight_int8_zero_check,
+            _linear_fp_act_uint4_weight_int8_zero_impl,
         ),
         (
             _linear_bf16_act_uint4_weight_float_zero_check,

Original file line number	Diff line number	Diff line change
`@@ -235,8 +235,8 @@ def _register_aqt_quantized_linear_dispatches():`
`235`	`235`	`_linear_q_dq_impl,`
`236`	`236`	`),`
`237`	`237`	`(`
`238`		`- _linear_bf16_act_uint4_weight_int8_zero_check,`
`239`		`- _linear_bf16_act_uint4_weight_int8_zero_impl,`
	`238`	`+ _linear_fp_act_uint4_weight_int8_zero_check,`
	`239`	`+ _linear_fp_act_uint4_weight_int8_zero_impl,`
`240`	`240`	`),`
`241`	`241`	`(`
`242`	`242`	`_linear_bf16_act_uint4_weight_float_zero_check,`