We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent be608fa commit e6f3056Copy full SHA for e6f3056
src/lightning/fabric/accelerators/xla.py
@@ -117,10 +117,6 @@ def _using_pjrt() -> bool:
117
118
return xr.using_pjrt()
119
120
- from torch_xla.experimental import pjrt
121
-
122
- return pjrt.using_pjrt()
123
124
125
def _parse_tpu_devices(devices: Union[int, str, list[int]]) -> Union[int, list[int]]:
126
"""Parses the TPU devices given in the format as accepted by the
0 commit comments