NVIDIA / NVIDIA/Megatron-LM

[QUESTION] In what cases will tp_comm_overlap accelerate training

Open
#2,662 5 comments 0 reactions 0 assignees View on GitHub
community-request question waiting-on-customer
Dominant language
Python
Stars
17.9k
Forks
4.5k
Avg merge
4d 6h
Merged PRs (30d)
271

Description

Using H20*8 to finetune Qwen3-32B with tp8 and 16384 packed_seq_len (with padding for fixed seq length as required by tp_comm_overlap userbuffer). The global batchsize is 16.

With tp_comm_overlap, it only speeds up by less than 2s/step from 43s/step, which is minor.

Are there any guides on for what model size, tp size and seq length will tp_comm_overlap be beneficial?

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.