Skip to content

Update AWS OFI NCCL version #3844

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

Open
wants to merge 11 commits into
base: main
Choose a base branch
from
Open

Update AWS OFI NCCL version #3844

wants to merge 11 commits into from

Conversation

dakinggg
Copy link
Contributor

@dakinggg dakinggg commented May 2, 2025

What does this PR do?

Updates to the latest OFI NCCL version

WIP retesting after comment from AWS that we don't need to separately install the ofi nccl plugin
Before: llama70b-8192-ofi-old-image-aws-1-0g5Lef
After: llama70b-8192-ofi-new-vendored-aws-2-a1gQZ4

throughput is slightly higher on the new image.

Note: we were originally going to go back to using latest for the EFA installer version, but 1.40 is producing an installation an error, so I'm pinning that to 1.39 (previous version). This is a known issue here with the debug install, so should be fixed soonish.

@dakinggg dakinggg marked this pull request as ready for review May 5, 2025 17:21
@dakinggg dakinggg requested a review from a team as a code owner May 5, 2025 17:21
Copy link
Member

@willgleich willgleich left a comment

Choose a reason for hiding this comment

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

lgtm thanks for the info regarding the EFA INSTALLER @rajachan

@dakinggg dakinggg marked this pull request as draft May 5, 2025 21:09
@dakinggg dakinggg marked this pull request as ready for review May 5, 2025 23:03
@dakinggg dakinggg requested a review from willgleich May 5, 2025 23:03
Sign up for free to 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.

3 participants