[BE] Make ScalingGranularity module level so it can be rendered in API ref on docsite (#2314)

danielvegamyhre · web-flow · commit 2ec325da0bff · 2025-06-11T09:18:37.000-07:00
diff --git a/torchao/float8/__init__.py b/torchao/float8/__init__.py
@@ -29,12 +29,14 @@
             GemmInputRole,
             LinearMMConfig,
             Float8MMConfig,
+            ScalingGranularity,
         ]
     )
 
 __all__ = [
     # configuration
     "ScalingType",
+    "ScalingGranularity",
     "Float8GemmConfig",
     "Float8LinearConfig",
     "CastConfig",
