bridge.perf_recipes.qwen.common#

Common helpers for qwen performance recipes.

Module Contents#

Functions#

_with_global_batch_size

_enable_hybridep_full_iteration_mxfp8

_enable_hybridep_full_iteration_nvfp4

Enable the full-iteration HybridEP stack for an NVFP4 benchmark.

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#
bridge.perf_recipes.qwen.common._enable_hybridep_full_iteration_nvfp4(
cfg: megatron.bridge.training.config.ConfigContainer,
) → None#

Enable the full-iteration HybridEP stack for an NVFP4 benchmark.