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

bionemo: bn2/add pipelineparallel dtype #9475

Merged

Conversation

skothenhill-nv
Copy link
Contributor

Fixes a bug from parallel_dtype not being correctly passed in the parallelism method.

@skothenhill-nv skothenhill-nv force-pushed the hillst-bn2/add-pipelineparallel-dtype branch from 2d472cc to e57a462 Compare June 14, 2024 16:42
Signed-off-by: skothenhill-nv <skothenhill-nv@users.noreply.github.com>
@marcromeyn marcromeyn self-requested a review June 14, 2024 17:01
@marcromeyn marcromeyn merged commit a5da602 into NVIDIA:main Jun 14, 2024
108 of 109 checks passed
JesusPaz pushed a commit to JesusPaz/NeMo that referenced this pull request Jun 18, 2024
* added pipeline_dtype for pipeline parallelism to megatron strategy and parallelism calls

* fix typos

* Apply isort and black reformatting

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

---------

Signed-off-by: skothenhill-nv <skothenhill-nv@users.noreply.github.com>
Co-authored-by: skothenhill-nv <skothenhill-nv@users.noreply.github.com>
rohitrango pushed a commit to rohitrango/NeMo that referenced this pull request Jun 25, 2024
* added pipeline_dtype for pipeline parallelism to megatron strategy and parallelism calls

* fix typos

* Apply isort and black reformatting

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

---------

Signed-off-by: skothenhill-nv <skothenhill-nv@users.noreply.github.com>
Co-authored-by: skothenhill-nv <skothenhill-nv@users.noreply.github.com>
@ko3n1g ko3n1g mentioned this pull request Jul 18, 2024
2 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants