这是indexloc提供的服务,不要输入任何密码
Skip to content

minor perf fixes #12656

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

Merged
merged 15 commits into from
Mar 21, 2025
Merged

minor perf fixes #12656

merged 15 commits into from
Mar 21, 2025

Conversation

malay-nagda
Copy link
Collaborator

@malay-nagda malay-nagda commented Mar 18, 2025

Important

The Update branch button must only be pressed in very rare occassions.
An outdated branch is never blocking the merge of a PR.
Please reach out to the automation team before pressing that button.

What does this PR do ?

  • assert HF_TOKEN is set when importing ckpt from HF
  • add TFLOPs/sec logging
  • fix num_train_samples to be exactly equal for 1 epoch
  • update seq len for b200 llama3 8b PEFT
  • updated 8x22b recipe to new 8x22b_64k recipe (the former used to have 64k seq len default)

Collection: [Note which collection this PR will affect]

Changelog

  • Add specific line by line info of high level changes in this PR.

Usage

  • You can potentially add a usage example below
# Add a code snippet demonstrating how to use this 

GitHub Actions CI

The Jenkins CI system has been replaced by GitHub Actions self-hosted runners.

The GitHub Actions CI will run automatically when the "Run CICD" label is added to the PR.
To re-run CI remove and add the label again.
To run CI on an untrusted fork, a NeMo user with write access must first click "Approve and run".

Before your PR is "Ready for review"

Pre checks:

  • Make sure you read and followed Contributor guidelines
  • Did you write any new necessary tests?
  • Did you add or update any necessary documentation?
  • Does the PR affect components that are optional to install? (Ex: Numba, Pynini, Apex etc)
    • Reviewer: Does the PR have correct import guards for all optional libraries?

PR Type:

  • New Feature
  • Bugfix
  • Documentation

If you haven't finished some of the above items you can still open "Draft" PR.

Who can review?

Anyone in the community is free to review the PR once the checks have passed.
Contributor guidelines contains specific people who can review PRs to various areas.

Additional Information

  • Related to # (issue)

Signed-off-by: Malay Nagda <malayn@nvidia.com>
Signed-off-by: Malay Nagda <malayn@nvidia.com>
Signed-off-by: Malay Nagda <malayn@nvidia.com>
Signed-off-by: Malay Nagda <malayn@nvidia.com>
Signed-off-by: Malay Nagda <malayn@nvidia.com>
Signed-off-by: Malay Nagda <malayn@nvidia.com>
@malay-nagda malay-nagda requested a review from erhoo82 March 18, 2025 15:13
@malay-nagda malay-nagda marked this pull request as ready for review March 18, 2025 15:13
malay-nagda and others added 3 commits March 19, 2025 22:33
Signed-off-by: Malay Nagda <malayn@nvidia.com>
Signed-off-by: malay-nagda <malay-nagda@users.noreply.github.com>
Signed-off-by: artbataev <artbataev@users.noreply.github.com>
malay-nagda and others added 5 commits March 19, 2025 22:45
Signed-off-by: Malay Nagda <malayn@nvidia.com>
Signed-off-by: malay-nagda <malay-nagda@users.noreply.github.com>
Signed-off-by: Malay Nagda <malayn@nvidia.com>
Signed-off-by: Malay Nagda <malayn@nvidia.com>
Signed-off-by: Malay Nagda <malayn@nvidia.com>
erhoo82
erhoo82 previously approved these changes Mar 20, 2025
Signed-off-by: malay-nagda <malayn@nvidia.com>
@malay-nagda malay-nagda enabled auto-merge (squash) March 21, 2025 06:42
@github-actions github-actions bot removed the Run CICD label Mar 21, 2025
Copy link
Contributor

[🤖]: Hi @malay-nagda 👋,

We wanted to let you know that a CICD pipeline for this PR just finished successfully

So it might be time to merge this PR or get some approvals

I'm just a bot so I'll leave it you what to do next.

//cc @pablo-garay @ko3n1g

@malay-nagda malay-nagda merged commit d26ad52 into main Mar 21, 2025
217 checks passed
@malay-nagda malay-nagda deleted the malay/minor_perf_fixes branch March 21, 2025 08:57
cspades pushed a commit that referenced this pull request Mar 24, 2025
* minor perf fixes

Signed-off-by: Malay Nagda <malayn@nvidia.com>

* flops callback to perf scripts

Signed-off-by: Malay Nagda <malayn@nvidia.com>

* no val loop

Signed-off-by: Malay Nagda <malayn@nvidia.com>

* mixtral kwargs extra arg

Signed-off-by: Malay Nagda <malayn@nvidia.com>

* mixtral kwargs extra arg

Signed-off-by: Malay Nagda <malayn@nvidia.com>

* mixtral kwargs extra arg

Signed-off-by: Malay Nagda <malayn@nvidia.com>

* single flops callback

Signed-off-by: Malay Nagda <malayn@nvidia.com>

* Apply isort and black reformatting

Signed-off-by: malay-nagda <malay-nagda@users.noreply.github.com>

* Apply isort and black reformatting

Signed-off-by: artbataev <artbataev@users.noreply.github.com>

* flux updates

Signed-off-by: Malay Nagda <malayn@nvidia.com>

* Apply isort and black reformatting

Signed-off-by: malay-nagda <malay-nagda@users.noreply.github.com>

* unused var

Signed-off-by: Malay Nagda <malayn@nvidia.com>

* add cg var

Signed-off-by: Malay Nagda <malayn@nvidia.com>

* moe 8x22b

Signed-off-by: Malay Nagda <malayn@nvidia.com>

---------

Signed-off-by: Malay Nagda <malayn@nvidia.com>
Signed-off-by: malay-nagda <malay-nagda@users.noreply.github.com>
Signed-off-by: artbataev <artbataev@users.noreply.github.com>
Signed-off-by: malay-nagda <malayn@nvidia.com>
Co-authored-by: malay-nagda <malay-nagda@users.noreply.github.com>
Co-authored-by: artbataev <artbataev@users.noreply.github.com>
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