Skip to content

Comments

bionemo: bn2/add pipelineparallel dtype#9475

Merged
marcromeyn merged 4 commits intoNVIDIA-NeMo:mainfrom
skothenhill-nv:hillst-bn2/add-pipelineparallel-dtype
Jun 14, 2024
Merged

bionemo: bn2/add pipelineparallel dtype#9475
marcromeyn merged 4 commits intoNVIDIA-NeMo:mainfrom
skothenhill-nv:hillst-bn2/add-pipelineparallel-dtype

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-NeMo:main Jun 14, 2024
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
XuesongYang pushed a commit to paarthneekhara/NeMo that referenced this pull request Jan 18, 2025
* 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>
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