Skip to content

Commit bb3b9a6

Browse files
odiemm-metafacebook-github-bot
authored andcommitted
add recipe config in aps for fp8
Summary: Add option for user to select scaling recipe in fp8 Differential Revision: D75732154
1 parent 488ecd4 commit bb3b9a6

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

torchao/float8/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
Float8GemmConfig,
55
Float8LinearConfig,
66
ScalingType,
7+
Float8LinearRecipeName,
78
)
89
from torchao.float8.float8_linear_utils import (
910
convert_to_float8_training,
@@ -38,6 +39,7 @@
3839
"Float8GemmConfig",
3940
"Float8LinearConfig",
4041
"CastConfig",
42+
"Float8LinearRecipeName",
4143
# top level UX
4244
"convert_to_float8_training",
4345
"precompute_float8_dynamic_scale_for_fsdp",

0 commit comments

Comments
 (0)