Skip to content

Pad sequences to multiples of 16 for GPTSFTDataset#7904

Merged
ericharper merged 2 commits intoNVIDIA-NeMo:mainfrom
vysarge:vsarge/pad16
Nov 17, 2023
Merged

Pad sequences to multiples of 16 for GPTSFTDataset#7904
ericharper merged 2 commits intoNVIDIA-NeMo:mainfrom
vysarge:vsarge/pad16

Conversation

@vysarge
Copy link
Contributor

@vysarge vysarge commented Nov 17, 2023

One-line change to pad sequences to the next highest multiple of 16 instead of 8 in GPTSFTDataset to allow this dataset's outputs to work with TransformerEngine which expects multiples of 16 when using fp8.

Signed-off-by: Valerie Sarge <vsarge@nvidia.com>
@github-actions github-actions bot added the NLP label Nov 17, 2023
@ericharper
Copy link
Collaborator

jenkins

Copy link
Collaborator

@ericharper ericharper 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!

@ericharper ericharper merged commit eba699d into NVIDIA-NeMo:main Nov 17, 2023
@vysarge vysarge deleted the vsarge/pad16 branch November 18, 2023 03:29
pzelasko pushed a commit to pzelasko/NeMo that referenced this pull request Jan 3, 2024
Signed-off-by: Valerie Sarge <vsarge@nvidia.com>
Co-authored-by: Eric Harper <complex451@gmail.com>
Signed-off-by: Piotr Żelasko <petezor@gmail.com>
rohitrango pushed a commit to rohitrango/NeMo that referenced this pull request Jun 25, 2024
Signed-off-by: Valerie Sarge <vsarge@nvidia.com>
Co-authored-by: Eric Harper <complex451@gmail.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