Rename Floating point to fp8 (#909)

pytorch · Sep 20, 2024 · 0bdde92 · 0bdde92
1 parent 4bce694
commit 0bdde92
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torchao/dtypes/affine_quantized_tensor.py b/torchao/dtypes/affine_quantized_tensor.py
@@ -1360,7 +1360,7 @@ def _linear_f16_act_floatx_weight_impl(input_tensor, weight_tensor, bias):
 
  return out.view(*act.shape[:-1], out_dim).to(act.dtype)
 
-def _linear_fp_act_fp8_weight_check(
+def _linear_fp8_act_fp8_weight_check(
  input_tensor: Union[torch.Tensor, AffineQuantizedTensor],
  weight_tensor: Union[torch.Tensor, AffineQuantizedTensor],
  bias: Optional[torch.Tensor],
@@ -1384,7 +1384,7 @@ def preprocess_scale(input_scale: torch.Tensor, input_shape: Tuple[int]):
 
  return input_scale
 
-def _linear_fp_act_fp8_weight_impl(
+def _linear_fp8_act_fp8_weight_impl(
  input_tensor: AffineQuantizedTensor,
  weight_tensor: AffineQuantizedTensor,
  bias: Optional[torch.Tensor],
@@ -1473,7 +1473,7 @@ def _register_aqt_quantized_linear_dispatches():
  for dispatch_condition, impl in [
  (_linear_int8_act_int8_weight_check, _linear_int8_act_int8_weight_impl),
  (_linear_int8_act_int8_weight_semi_structured_sparse_check, _linear_int8_act_int8_weight_semi_structured_sparse_impl),
- (_linear_fp_act_fp8_weight_check, _linear_fp_act_fp8_weight_impl),
+ (_linear_fp8_act_fp8_weight_check, _linear_fp8_act_fp8_weight_impl),
  (_linear_bf16_act_uint4_weight_check, _linear_bf16_act_uint4_weight_impl),
  (_linear_fp_act_int8_weight_check, _linear_fp_act_int8_weight_impl),
  (_linear_f16_act_floatx_weight_check, _linear_f16_act_floatx_weight_impl),