Skip to content

Add checks for tensor parallel use case to ensure all-reduce is called only when necessary - #3

Merged
ptrendx merged 1 commit into
NVIDIA:mainfrom
ksivaman:tensor_parallel_fix
Oct 4, 2022
Merged

Add checks for tensor parallel use case to ensure all-reduce is called only when necessary#3
ptrendx merged 1 commit into
NVIDIA:mainfrom
ksivaman:tensor_parallel_fix

Conversation

@ksivaman

Copy link
Copy Markdown
Member

Signed-off-by: Kirthi Shankar Sivamani ksivamani@nvidia.com

Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

@ptrendxptrendx left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@ptrendx
ptrendx merged commit 5612ba7 into NVIDIA:mainOct 4, 2022
@ksivaman
ksivaman deleted the tensor_parallel_fix branch July 19, 2023 01:39
@klhhhhhklhhhhh mentioned this pull request Nov 4, 2024
@greptile-appsgreptile-appsBot mentioned this pull request Jan 12, 2026
13 tasks
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants

@ksivaman@ptrendx