Remove or simplify hardcoded lists of device types #6235
Remove or simplify some device type lists
2fbe5283
remove instances of `GPU` and `ROCM` devices
2fa33286
Add `PLUGIN` device type
f4ec98b1
fix `get_xla_supported_devices`
ab627c43
formatting
8c9c6d4b
remove old GPU conditions
65f55284
Remove XPU cases
b9e77563
enumerate
2026f4d2
GPU -> CUDA
02b72270
Merge branch 'master' of github.com:pytorch/xla into wcromar/simplify…
264e1e2d
Merge branch 'master' of github.com:pytorch/xla into wcromar/simplify…
7809ef78
Store device type name in `torch_xla::DeviceType`
a014da97
formatting
6d978788
Add `ComputationClient::GetDeviceType`
80ad8afd
clarify skip conditions
71e6e52e
will-cromar
changed the title [WIP] Remove or simplify hardcoded lists of device types Remove or simplify hardcoded lists of device types 2 years ago
Merge branch 'master' of github.com:pytorch/xla into wcromar/simplify…
430ab6bf
will-cromar
marked this pull request as ready for review 2 years ago
JackCaoG
approved these changes
on 2024-01-10
Merge branch 'master' of github.com:pytorch/xla into wcromar/simplify…
02b5d4de
remove useless constructor
96934e7e
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub