pytorch/torch/cpu
wz337 a614281ea9 Add current_device() to torch.cpu (#110987)
Better support device agnostic, add a "cpu" return for `current_device()` in torch.cpu so that we won't run into `AttributeError: module 'torch.cpu' has no attribute 'current_device'`.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/110987
Approved by: https://github.com/wanchaol
2023-10-11 05:13:10 +00:00
..
amp Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00
__init__.py Add current_device() to torch.cpu (#110987) 2023-10-11 05:13:10 +00:00