Code

Tpu save#4309

Code

Merged

SeanNaren merged 27 commits intomasterLightning-AI/pytorch-lightning:masterfrom tpu_saveLightning-AI/pytorch-lightning:tpu_saveCopy head branch name to clipboardDec 2, 2020

+172-98

Commits

Commits on Dec 2, 2020

• convert xla tensor to cpu before save

  

  lezwon

  authored and

  Borda

  committed
  c900608

  Copy full SHA for c900608

• move_to_cpu

  

  lezwon

  authored and

  Borda

  committed
  8c1e3f7

  Copy full SHA for 8c1e3f7

• updated CHANGELOG.md

  

  lezwon

  authored and

  Borda

  committed
  654c15b

  Copy full SHA for 654c15b

• added on_save to accelerators

  

  lezwon

  authored and

  Borda

  committed
  031af97

  Copy full SHA for 031af97

• if accelerator is not None

  

  lezwon

  authored and

  Borda

  committed
  00256b8

  Copy full SHA for 00256b8

• refactors

  

  lezwon

  authored and

  Borda

  committed
  8fcf157

  Copy full SHA for 8fcf157

• change filename to run test

  

  lezwon

  authored and

  Borda

  committed
  a9a8188

  Copy full SHA for a9a8188

• run test_tpu_backend

  

  lezwon

  authored and

  Borda

  committed
  9e78274

  Copy full SHA for 9e78274

• added xla_device_utils to tests

  

  lezwon

  authored and

  Borda

  committed
  5becb7b

  Copy full SHA for 5becb7b

• added xla_device_utils to test

  

  lezwon

  authored and

  Borda

  committed
  1946591

  Copy full SHA for 1946591

• removed tests

  

  lezwon

  authored and

  Borda

  committed
  502e8a1

  Copy full SHA for 502e8a1

• Revert "added xla_device_utils to test"

  Show description for c6e4535
  

  lezwon

  authored and

  Borda

  committed
  c6e4535

  Copy full SHA for c6e4535

• fixed pep

  

  lezwon

  authored and

  Borda

  committed
  3a35796

  Copy full SHA for 3a35796

• increase timeout and print traceback

  

  lezwon

  authored and

  Borda

  committed
  0565e93

  Copy full SHA for 0565e93

• lazy check tpu exists

  

  lezwon

  authored and

  Borda

  committed
  7321045

  Copy full SHA for 7321045

• increased timeout

  Show description for 5573d51
  

  lezwon

  authored and

  Borda

  committed
  5573d51

  Copy full SHA for 5573d51

• fixed torch_xla imports

  

  lezwon

  authored and

  Borda

  committed
  da59b4d

  Copy full SHA for da59b4d

• fix tests

  

  lezwon

  authored and

  Borda

  committed
  63c6e15

  Copy full SHA for 63c6e15

• define xla utils

  

  lezwon

  authored and

  Borda

  committed
  57029e3

  Copy full SHA for 57029e3

• fix test

  

  lezwon

  authored and

  Borda

  committed
  b34d615

  Copy full SHA for b34d615

• aval

  Borda

  committed
  9c0c567

  Copy full SHA for 9c0c567

• chlog

  Borda

  authored
  e4eb7ed

  Copy full SHA for e4eb7ed

• docs

  Borda

  committed
  d06c1b2

  Copy full SHA for d06c1b2

• aval

  Borda

  committed
  b180798

  Copy full SHA for b180798

• Apply suggestions from code review

  Borda

  authored
  bab6d12

  Copy full SHA for bab6d12

• Merge branch 'master' into tpu_save

  tchaton

  authored
  a485e29

  Copy full SHA for a485e29

• Merge branch 'master' into tpu_save

  SeanNaren

  authored
  eef0849

  Copy full SHA for eef0849