Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[pull] main from NVIDIA:main #50

Merged
merged 2 commits into from
Jan 14, 2025
Merged

[pull] main from NVIDIA:main #50

merged 2 commits into from
Jan 14, 2025

Conversation

pull[bot]
Copy link

@pull pull bot commented Jan 14, 2025

See Commits and Changes for more details.


Created by pull[bot] (v2.0.0-alpha.1)

Can you help keep this open source service alive? 💖 Please sponsor : )

kit1980 and others added 2 commits January 13, 2025 11:27
…de="column"` (#1343)

* support AG overlap in sequence-parallel Linear forward and RS overlap in sequence-parallel Linear backward

Signed-off-by: Alp Dener <[email protected]>

* implemented TP overlap support for column-parallel te.Linear

Signed-off-by: Alp Dener <[email protected]>

* fixed backward pass for te.Linear column-parallel with TP overlap, updated unit tests

Signed-off-by: Alp Dener <[email protected]>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* improved error messages for internal failure to infer TP overlap options in te.Linear

Signed-off-by: Alp Dener <[email protected]>

* fixed linting errors

Signed-off-by: Alp Dener <[email protected]>

* fixed incorrect TP overlap option asserts

Signed-off-by: Alp Dener <[email protected]>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Signed-off-by: Alp Dener <[email protected]>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
@pull pull bot added the ⤵️ pull label Jan 14, 2025
@pull pull bot merged commit 2402406 into phu0ngng:main Jan 14, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants