Skip to content

Add NeMo framework example to a3-Ultra#3477

Merged
tpdownes merged 1 commit into
GoogleCloudPlatform:developfrom
akiki-liang0:feat/a3ultra-preview-add-nemo
Jan 10, 2025
Merged

Add NeMo framework example to a3-Ultra#3477
tpdownes merged 1 commit into
GoogleCloudPlatform:developfrom
akiki-liang0:feat/a3ultra-preview-add-nemo

Conversation

@akiki-liang0

Copy link
Copy Markdown
Contributor
  • Added appropriate environment variables, container mounts, and example command for running NeMo on a3ultra
  • ensure NCCL_NET=gIB is being used

Testing:

  • 10 consecutive successful runs using example command in README.md

@akiki-liang0 akiki-liang0 reopened this Dec 30, 2024
@akiki-liang0 akiki-liang0 changed the base branch from main to a3ultra-preview December 30, 2024 22:19
@tpdownes tpdownes force-pushed the a3ultra-preview branch 2 times, most recently from 6374d7d to 7cd0a0e Compare December 30, 2024 23:19
@akiki-liang0 akiki-liang0 changed the base branch from a3ultra-preview to develop January 10, 2025 21:00
@tpdownes tpdownes force-pushed the feat/a3ultra-preview-add-nemo branch from 0671733 to 9b9bd8c Compare January 10, 2025 21:26
@tpdownes tpdownes added the release-key-new-features Added to release notes under the "Key New Features" heading. label Jan 10, 2025
@tpdownes tpdownes self-assigned this Jan 10, 2025
@tpdownes tpdownes self-requested a review January 10, 2025 21:43

@tpdownes tpdownes left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM. Thanks for adding testing information.

@tpdownes

Copy link
Copy Markdown
Contributor

/gcbrun

@tpdownes tpdownes enabled auto-merge January 10, 2025 23:20
@tpdownes tpdownes merged commit ef16361 into GoogleCloudPlatform:develop Jan 10, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

release-key-new-features Added to release notes under the "Key New Features" heading.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants