Skip to content

Conversation

@zhiqiu
Copy link
Contributor

@zhiqiu zhiqiu commented Jan 15, 2021

PR types

Function optimization

PR changes

Others

Describe

Unify calling cudaSetDevice in paddle (except code in fleetX)

cudaSetDevice(dev_id) -> platform::SetDeviceId(dev_id);

  • Advantage
    provide unified and better error message and logging for debug

@paddle-bot-old
Copy link

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

@zhiqiu zhiqiu requested a review from chenwhql January 15, 2021 03:59
@zhiqiu zhiqiu force-pushed the dev/unify_set_device_id branch from 379f973 to 265db22 Compare January 15, 2021 08:06
@zhiqiu zhiqiu merged commit 81217a9 into PaddlePaddle:develop Jan 19, 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.

2 participants