`bridge.perf_recipes.nemotronh.common`#

Common helpers for nemotronh performance recipes.

Module Contents#

Functions#

`_with_global_batch_size`
`_nemotron_3_super_nvfp4_precision`	Return the NVFP4 precision config used by Nemotron 3 Super perf recipes.
`_apply_nemotron_3_super_perf_defaults`	Apply shared Nemotron 3 Super perf defaults after recipe-specific overrides.

Data#

_TE_QUANT_CFG_PATH

API#

bridge.perf_recipes.nemotronh.common._TE_QUANT_CFG_PATH#: ‘with_name(…)’

bridge.perf_recipes.nemotronh.common._with_global_batch_size( cfg: megatron.bridge.training.config.ConfigContainer, global_batch_size: int, ) → megatron.bridge.training.config.ConfigContainer#

bridge.perf_recipes.nemotronh.common._nemotron_3_super_nvfp4_precision() → megatron.bridge.training.mixed_precision.MixedPrecisionConfig#: Return the NVFP4 precision config used by Nemotron 3 Super perf recipes.

bridge.perf_recipes.nemotronh.common._apply_nemotron_3_super_perf_defaults( cfg: megatron.bridge.training.config.ConfigContainer, ) → None#: Apply shared Nemotron 3 Super perf defaults after recipe-specific overrides.