From 99848528c33baae0b21564c18d847d76d5f2a5f6 Mon Sep 17 00:00:00 2001
From: Xue Dong <xued@meta.com>
Date: Thu, 5 Jun 2025 15:28:18 -0700
Subject: [PATCH] add recipe config in aps for fp8 (#2322)

Summary:

Add option for user to select scaling recipe in fp8

Differential Revision: D75732154
---
 torchao/float8/__init__.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/torchao/float8/__init__.py b/torchao/float8/__init__.py
index 8d4d58fd6b..c8812e29ec 100644
--- a/torchao/float8/__init__.py
+++ b/torchao/float8/__init__.py
@@ -4,6 +4,7 @@
     Float8GemmConfig,
     Float8LinearConfig,
     ScalingType,
+    Float8LinearRecipeName,
 )
 from torchao.float8.float8_linear_utils import (
     convert_to_float8_training,
@@ -38,6 +39,7 @@
     "Float8GemmConfig",
     "Float8LinearConfig",
     "CastConfig",
+    "Float8LinearRecipeName",
     # top level UX
     "convert_to_float8_training",
     "precompute_float8_dynamic_scale_for_fsdp",