pytorch
9024015a - [BE] Small improvements to device_count (#85192)

Commit
2 years ago
[BE] Small improvements to device_count (#85192) If `_parse_visible_devices` returns an empty set, no need to make nvml calls Also, reduce indent a bit in `_device_count_nvml` Pull Request resolved: https://github.com/pytorch/pytorch/pull/85192 Approved by: https://github.com/kit1980, https://github.com/ngimel
Author
Committer
Parents
Loading