Skip to content

FP8

以下是与底层 FP8 实现相关的函数和类

FP8RecipeKwargs

[[autodoc]] utils.FP8RecipeKwargs

convert_model

[[autodoc]] utils.convert_model

has_transformer_engine_layers

[[autodoc]] utils.has_transformer_engine_layers

contextual_fp8_autocast

[[autodoc]] utils.contextual_fp8_autocast

apply_fp8_autowrap

[[autodoc]] utils.apply_fp8_autowrap