pytorch
c318de42 - [dynamo] Get GPU names without calling nvidia-smi (#90474)

Commit
2 years ago
[dynamo] Get GPU names without calling nvidia-smi (#90474) Believe it or not, inductor can sometimes be used on machines that have CUDA GPUs but no nvidia-smi. Let's use torch APIs instead of subprocess. Differential Revision: [D41841930](https://our.internmc.facebook.com/intern/diff/D41841930/) Differential Revision: [D41841930](https://our.internmc.facebook.com/intern/diff/D41841930) Pull Request resolved: https://github.com/pytorch/pytorch/pull/90474 Approved by: https://github.com/voznesenskym, https://github.com/anijain2305
Author
Committer
Parents
Loading