NVIDIA / NVIDIA/Megatron-LM

[QUESTION] Why A2A Overlap encounters hang with torch version < 2.6.0

Open
#1,828 1 comment 0 reactions 0 assignees View on GitHub
bug
Dominant language
Python
Stars
17.9k
Forks
4.5k
Avg merge
4d 6h
Merged PRs (30d)
271

Description

I'm seeing `A2A Overlap encounters hang issue with torch version < 2.6.0` in the code. I'd like to know why this is happening. I ran it with torch 2.5 and it doesn't seem to hang.

Appreciate any insights or help.

@lhb8125

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.