We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 488ecd4 commit bb3b9a6Copy full SHA for bb3b9a6
torchao/float8/__init__.py
@@ -4,6 +4,7 @@
4
Float8GemmConfig,
5
Float8LinearConfig,
6
ScalingType,
7
+ Float8LinearRecipeName,
8
)
9
from torchao.float8.float8_linear_utils import (
10
convert_to_float8_training,
@@ -38,6 +39,7 @@
38
39
"Float8GemmConfig",
40
"Float8LinearConfig",
41
"CastConfig",
42
+ "Float8LinearRecipeName",
43
# top level UX
44
"convert_to_float8_training",
45
"precompute_float8_dynamic_scale_for_fsdp",
0 commit comments