pytorch-lightning
Replace `using_pjrt()` xla runtime `device_type()` check with in xla.py for `torch-xla>=2.5`
#20442

Merged

Replace `using_pjrt()` xla runtime `device_type()` check with in xla.py for `torch-xla>=2.5` #20442

lantiga merged 6 commits into Lightning-AI:master from byi8220:patch-1

Replace `using_pjrt()` xla runtime `device_type()` check with in xla.py

da605799

github-actions added fabric

Added tests/refactored for version compat

5c659320

Version compat

012e18af

[pre-commit.ci] auto fixes from pre-commit.com hooks

d1cf1009

precommit

bd184868

merge conflicts

5ada0e1e

byi8220 changed the title ~~Replace `using_pjrt()` xla runtime `device_type()` check with in xla.py~~ Replace `using_pjrt()` xla runtime `device_type()` check with in xla.py for `torch-xla>=2.5` 1 year ago

byi8220 marked this pull request as ready for review 1 year ago