NVIDIA / NVIDIA/Megatron-LM

Change LayerwiseDistributedOptimizer's Allgather_v into Allgather

Open
#3,178 1 comment 0 reactions 1 assignee Claimed by @FDecaYed View on GitHub
enhancement
Dominant language
Python
Stars
17.9k
Forks
4.5k
Avg merge
4d 6h
Merged PRs (30d)
271

Description

Currently LayerwiseDistributedOptimizer is using AG_v, we can change it to AG with paddings. Implement and analyze this change

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.