bridge.perf_recipes.wan.h100.wan#

H100 performance recipes for Wan.

Module Contents#

Functions#

wan_14b_pretrain_32gpu_h100_bf16_config

Wan 14B pretrain: 32× H100, BF16, TP=2 CP=4, recompute block/8.

API#

bridge.perf_recipes.wan.h100.wan.wan_14b_pretrain_32gpu_h100_bf16_config() megatron.bridge.perf_recipes.wan.common.ConfigContainer#

Wan 14B pretrain: 32× H100, BF16, TP=2 CP=4, recompute block/8.