lint

Author: Peter Y. Yeh

test/quantization/test_quant_api.py

@@ -55,8 +55,8 @@
     TORCH_VERSION_AT_LEAST_2_6,
     is_sm_at_least_89,
     is_sm_at_least_90,
-    unwrap_tensor_subclass,
     skip_if_rocm,
+    unwrap_tensor_subclass,
 )
 
 try:

torchao/float8/config.py

@@ -148,7 +148,6 @@ class Float8GemmConfig:
 
 # Pre-made recipes for common configurations
 class Float8LinearRecipeName(enum.Enum):
-
     # Default, dynamic per-tensor scaling with the cuBLAS tensorwise kernel
     TENSORWISE = "tensorwise"
 
@@ -385,7 +384,6 @@ def from_recipe_name(
             )
 
         elif recipe_name is Float8LinearRecipeName.ROWWISE_WITH_GW_HP:
-
             # output_hp = input_fp8_axiswise_dim0 @ weight_t_axiswise_dim1
             cc_i = CastConfig(scaling_granularity=ScalingGranularity.AXISWISE)
             cc_w = CastConfig(scaling_granularity=ScalingGranularity.AXISWISE)