Chris Chow
|
f14a47a0b2
|
guard against None in pytorch get_xla_supported_devices (#9572)
Co-authored-by: Chris Chow <cchow@nianticlabs.com>
Co-authored-by: thomas chaton <thomas@grid.ai>
|
2021-10-12 12:01:32 +00:00 |
|
Jirka Borovec
|
6e124e7207
|
CI: precommit - docformatter (#8584)
* CI: precommit - docformatter
* fix deprecated
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2021-09-06 12:49:09 +00:00 |
|
Daniel Stancl
|
d063059d03
|
Fix mypy for `utilities.xla_device` (#8755)
* Fix mypy for utilities.xla_device
* Add explicit type hint for _XLA_AVAILABLE in utilities.imports
|
2021-08-06 17:01:21 +02:00 |
|
Carlos Mocholí
|
a64cc37394
|
Replace `yapf` with `black` (#7783)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2021-07-26 13:37:35 +02:00 |
|
Kaushik B
|
27eb0035ca
|
Increase TPU Check timeout (#7706)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
|
2021-05-26 01:44:29 +00:00 |
|
Kaushik B
|
e7057d5898
|
Add `should_rank_save_checkpoint` property to Training Plugins (#7684)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2021-05-25 23:02:05 +05:30 |
|
Kaushik B
|
bf46730d92
|
Support TPU Pod Training (n/n) (#7296)
|
2021-05-17 11:33:44 +00:00 |
|
Kaushik B
|
490cc57809
|
Device updates for TPU Pod (#7243)
|
2021-04-30 23:14:06 +05:30 |
|
thomas chaton
|
e147127c0e
|
[feat] Add better support for predict + ddp 2/3 (#7215)
* wip
* update
* update
* update
* update
* update
* typo
* update on comments
* update
* update
* update
* update
* update changelog
* update
* Fix merge
* Fix merge
* move code
* resolve test
* add extra test
* add an extra test
* update on comments
* add typing
* resolve flake8
* Refactor and Docs
* Fix tests
* Fix tests
* Fix tests
* Duplicate
* Fix tests
* resolve bug
* update
* update on comments
* update
* update changelog
* update
* update
* remove tpu
* resolve flake8
* update on comments
* update on comments
* update on comment
* resolve flake8
* add a cpu test for predict
* add None test
* update
* Update CHANGELOG.md
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
* resolve tests
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
|
2021-04-27 08:46:45 -04:00 |
|
Kaushik B
|
13f67ad313
|
Update logic for checking TPUs availability (#6767)
* Update logic for checking TPUs availability
* fix flake8
* add fix
|
2021-04-01 03:04:33 +05:30 |
|
thomas chaton
|
3a4c4246ee
|
[TPU] update is_tpu_exists utils internal logic to rely on xmp.spawn (#6719)
* update_logic
* update
* Update tests/utilities/test_xla_device_utils.py
* Update pytorch_lightning/utilities/xla_device.py
Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
* Update pytorch_lightning/utilities/xla_device.py
Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
* update test
* Update tests/utilities/test_xla_device_utils.py
* update
* Apply fix
* Docstring
* flake8
* update
Co-authored-by: Your Name <you@example.com>
Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
|
2021-03-29 18:59:20 +01:00 |
|
Lezwon Castelino
|
b95471d4a4
|
Increase TPU check timeout (#5598)
* change timeout to 100
* add to CHANGELOG.md
* update test
* updates
* reduce TPU_TIMEOUT_CONSTANT during test
* Update tests/utilities/test_xla_device_utils.py
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
* patch TPU_TIMEOUT_CONSTANT
* Apply suggestions from code review
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
|
2021-02-04 20:55:41 +01:00 |
|
Jirka Borovec
|
eee38d59e7
|
formatting to PL utils (#5713)
* yapf pl base
* over
* dist
* utils
* Apply suggestions from code review
* flake8
* neew way
|
2021-01-30 15:28:59 +01:00 |
|
Jirka Borovec
|
957583544a
|
mark todo exceptions (#5320)
* mark todo exceptions
* .
* .
* .
* .
* .
* .
* .
* .
* try
* .
|
2021-01-04 09:07:56 +01:00 |
|
Jirka Borovec
|
ae9956f997
|
clean avail. imports & enums (#5256)
* clean avail. imports
* enums
* fix missing
|
2020-12-29 19:02:18 +01:00 |
|
Jirka Borovec
|
a884866ff0
|
Unify names in Utils (#5199)
* warnings
* argparse
* mutils
* xla device
* deprecated
* tests
* simple
* flake8
* fix
* flake8
* 1.4
|
2020-12-22 00:23:33 +01:00 |