bridge.perf_recipes.qwen.common

Common helpers for qwen performance recipes.

Module Contents

Functions

_with_global_batch_size
_enable_hybridep_full_iteration_mxfp8

API

bridge.perf_recipes.qwen.common._with_global_batch_size(

cfg: megatron.bridge.training.config.ConfigContainer,

global_batch_size: int,

) → megatron.bridge.training.config.ConfigContainer

bridge.perf_recipes.qwen.common._enable_hybridep_full_iteration_mxfp8(

cfg: megatron.bridge.training.config.ConfigContainer,

) → None