Skip to content

Fix pin_memory to support single-element batch#1158

Closed
ejguan wants to merge 2 commits intometa-pytorch:mainfrom
ejguan:fix_single_batch_pin_memory
Closed

Fix pin_memory to support single-element batch#1158
ejguan wants to merge 2 commits intometa-pytorch:mainfrom
ejguan:fix_single_batch_pin_memory

Conversation

@ejguan
Copy link
Contributor

@ejguan ejguan commented May 12, 2023

Fixes #1157

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label May 12, 2023
@facebook-github-bot
Copy link
Contributor

@ejguan has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

@ejguan ejguan requested a review from NivekT May 12, 2023 14:24
Copy link
Contributor

@NivekT NivekT left a comment

Choose a reason for hiding this comment

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

LGTM! Just to confirm, I don't think this will change the result for non-single element case. Correct?

@ejguan
Copy link
Contributor Author

ejguan commented May 12, 2023

LGTM! Just to confirm, I don't think this will change the result for non-single element case. Correct?

Yes

@facebook-github-bot
Copy link
Contributor

@ejguan has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

@facebook-github-bot
Copy link
Contributor

@ejguan merged this pull request in 65e2ede.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. Merged

Projects

None yet

Development

Successfully merging this pull request may close these issues.

default pin_memory_fn with string Sequences and batch_size 1

3 participants