Use device index as zero if there is only one device
Open
banitag1
opened this issue 2 years ago
•
22 comments
Summary: We are seeing that many benchmark use "cuda" instead of "cuda:0", so index is not properly set. This diff adds a simple change to set the target index as zero if there is only one cuda device.
Differential Revision: D49001887
✅ Deploy Preview for pytorch-fbgemm-docs canceled.
Name
Link
🔨 Latest commit43e444f7f058ca936fd15aff73d212ec25bbada5
🔍 Latest deploy loghttps://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/64f8c452ea906200085edef2
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
This pull request was exported from Phabricator. Differential Revision: D49001887
pytorch
![netlify[bot] avatar](https://avatars.githubusercontent.com/in/13473?v=4 "Published by a pub.dev verified publisher"){kind=small}
