Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[TPU] Correct the check for TPU device in a pod environment #6755

Closed
wants to merge 12 commits into from
Closed

[TPU] Correct the check for TPU device in a pod environment #6755

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
fd4f471
fixup in a v2-32 env
jiasenwu Mar 30, 2021
01c274c
try
jiasenwu Mar 30, 2021
87ebd31
remove global flag _TPU_AVAILABLE
jiasenwu Mar 30, 2021
69021ce
timeout 25 => 60
jiasenwu Mar 30, 2021
be31c1f
index is required
jiasenwu Mar 30, 2021
5385fb0
resolved bad merge
jiasenwu Mar 30, 2021
9fef5a9
timeout 60 => 120
jiasenwu Mar 30, 2021
b44722b
replace _TPU_AVAILABLE in tests
jiasenwu Mar 31, 2021
9fd8ec2
remove comment and unused var
jiasenwu Mar 31, 2021
f804e6b
remove _TPU_AVAILABLE in doc
jiasenwu Mar 31, 2021
9cb858e
revert all changes to _TPU_AVAILABLE flag
jiasenwu Mar 31, 2021
2c5a700
one more revert
jiasenwu Mar 31, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
remove comment and unused var
  • Loading branch information
@jiasenwu
jiasenwu committed Mar 31, 2021
commit 9fd8ec2eb465337f46a7d3f8265e8c836789dc5a
3 changes: 1 addition & 2 deletions pytorch_lightning/utilities/xla_device.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -76,8 +76,7 @@ def _is_device_tpu(index) -> bool:
device = xm.xla_device()
return device.type == 'xla'

# Missing XLA Configuration
except RuntimeError as e:
except RuntimeError:
traceback.print_exc()
return False

Expand Down