Skip to content

Commit e6f3056

Browse files
authored
Update xla.py
1 parent be608fa commit e6f3056

File tree

1 file changed

+0
-4
lines changed
  • src/lightning/fabric/accelerators

1 file changed

+0
-4
lines changed

src/lightning/fabric/accelerators/xla.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -117,10 +117,6 @@ def _using_pjrt() -> bool:
117117

118118
return xr.using_pjrt()
119119

120-
from torch_xla.experimental import pjrt
121-
122-
return pjrt.using_pjrt()
123-
124120

125121
def _parse_tpu_devices(devices: Union[int, str, list[int]]) -> Union[int, list[int]]:
126122
"""Parses the TPU devices given in the format as accepted by the

0 commit comments

Comments
 (0)