Adrian Wälchli
0f9156374d
Mark internal Lite APIs as protected ( #15307 )
...
* mark internal lite apis as protected
* formatting
* docs update
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-26 12:51:50 +00:00
Jirka Borovec
d0b092fda8
Lite: setting extras & fix CI ( #15192 )
...
* extras
* test.txt
* doctest
* Apply suggestions from code review
* Fix imports
* Oops
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2022-10-19 19:05:23 +00:00
Carlos Mocholí
da25d1d30d
Remove unused Lite code ( #15000 )
...
* Remove unused Lite code
* Remove duplicate import
* Group variable
* Fix monkeypatch
2022-10-10 22:16:56 +00:00
Carlos Mocholí
7ef87464dd
Refactor XLA and TPU checks across codebase ( #14550 )
2022-10-04 22:54:14 +00:00
Carlos Mocholí
6256a318d7
Refactor launching tests to use our launchers ( #14954 )
2022-09-30 09:57:18 +02:00
Adrian Wälchli
d7404c775a
Integration tests for Precision in Lite ( #14815 )
...
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com>
2022-09-26 18:50:11 +00:00
Adrian Wälchli
dc1dc0df36
Attempt to query device count via NVML ( #14631 )
...
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2022-09-22 09:57:13 +00:00
otaj
5ee2b86c44
Tests for fixed TypeError ( #14821 )
...
* tests for 14809
* Apply suggestions from code review
Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-09-22 09:04:27 +02:00
Carlos Mocholí
7e803ba53e
Clean-up dtype management ( #14823 )
...
Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com>
2022-09-22 00:07:36 +00:00
Carlos Mocholí
e9c571d39f
Move accelerator-specific parsing functions with their accelerators ( #14753 )
...
Co-authored-by: awaelchli <aedu.waelchli@gmail.com>
2022-09-18 22:48:45 +00:00
Adrian Wälchli
deca6cc5c4
Standalone Lite: DDP Strategy Family ( #14670 )
...
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2022-09-15 01:36:17 +00:00
Adrian Wälchli
d013bcc5bf
Standalone Lite: Accelerators ( #14578 )
...
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2022-09-12 16:00:14 +00:00
Adrian Wälchli
95374440ce
Move device parser tests inside Lite ( #14586 )
2022-09-07 21:22:46 +00:00
Adrian Wälchli
d2459df2ff
Standalone Lite: Remaining Utilities ( #14492 )
...
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Laverne Henderson <laverne.henderson@coupa.com>
Co-authored-by: Felonious-Spellfire <felonious.spellfire@gmail.com>
2022-09-07 15:25:23 +00:00
Carlos Mocholí
273a9ed8c1
Integrate `lightning_utilities.core.apply_func` ( #14537 )
2022-09-06 13:52:54 +00:00
Carlos Mocholí
8a4a3b6766
Mark the lite `DeviceDtypeModuleMixin` as protected ( #14548 )
2022-09-06 14:17:15 +02:00
awaelchli
9fea2ed9d5
move pl/utilities/apply_func.py to pl/utilities/apply_func.py ( #14516 )
2022-09-05 20:30:42 +02:00
awaelchli
def6548596
move pl/utilities/cloud_io.py to lite/utilities/cloud_io.py ( #14515 )
2022-09-05 18:30:31 +02:00
awaelchli
75d5a2d046
move pl/utilities/xla_device.py to lite/utilities/xla_device.py ( #14514 )
2022-09-05 17:36:02 +02:00
awaelchli
cefe2fa123
Move test_dtype_device_mixin to lite ( #14511 )
2022-09-05 16:31:00 +02:00