Skip to content

Conversation

@zhiqiu
Copy link
Contributor

@zhiqiu zhiqiu commented Jan 13, 2021

PR types

Bug fixes

PR changes

Others

Describe

[cherry-pick] Set expected place in child thread for dataloader #30383

…da memory on other card (PaddlePaddle#30338)

* set expected place in child thread for dataloader

* set device id when set tensor from numpy

* revert tensor_py change

* add compile guard

* fix ci

* fix bug
@paddle-bot-old
Copy link

Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

@paddle-bot-old
Copy link

paddle-bot-old bot commented Jan 13, 2021

✅ This PR's description meets the template requirements!
Please wait for other CI results.

@zhiqiu zhiqiu changed the title [cherry-pick] Set expected place in child thread for dataloader to avoid costing cu… [cherry-pick] Set expected place in child thread for dataloader Jan 13, 2021
Copy link
Contributor

@XiaoguangHu01 XiaoguangHu01 left a comment

Choose a reason for hiding this comment

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

LGTM

@lanxianghit lanxianghit merged commit 9fb5a3e into PaddlePaddle:release/2.0 Jan 13, 2021
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