Skip to content

Commit

Permalink
Fix ruff for torchao/float8/config.py (#1750)
Browse files Browse the repository at this point in the history
  • Loading branch information
cpuhrsch authored Feb 21, 2025
1 parent 6bab4db commit 1c76736
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions torchao/float8/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"

Expand Down Expand Up @@ -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)
Expand Down

0 comments on commit 1c76736

Please sign in to comment.