3 Commits

Author	SHA1	Message	Date
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
Wanchao Liang	28d7d7fc42	device agnostic: torch.cpu.set_device (#110716) ... to support device agnostic, add a dummpy placeholder in torch.cpu Pull Request resolved: https://github.com/pytorch/pytorch/pull/110716 Approved by: https://github.com/albanD	2023-10-09 23:00:15 +00:00
Rodrigo Kumpera	fc012d716d	[core] Bring cpu device module closer to cuda's. (#103172) ... By implementing some of the functionality used by CUDA we make implementing device agnostic code a lot easier. With this set of changes it's now possible to get FSDP wrap a trivial module. FWD/BWD still TBD. Pull Request resolved: https://github.com/pytorch/pytorch/pull/103172 Approved by: https://github.com/wz337, https://github.com/wanchaol	2023-07-12 19:43:22 +00:00