Carlos Mocholí
|
7ef87464dd
|
Refactor XLA and TPU checks across codebase (#14550)
|
2022-10-04 22:54:14 +00:00 |
|
Justin Goodwin
|
45ca78167e
|
Improving Hydra+DDP support (#11617)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: rohitgr7 <rohitgr1998@gmail.com>
Co-authored-by: Jirka <jirka.borovec@seznam.cz>
|
2022-09-22 16:03:13 +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 |
|
Adrian Wälchli
|
fafd254678
|
Fix device parser logic to avoid creating CUDA context (#14319)
* let environment disable forking
* add helper function and error messages
* tests
* changelog
Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
|
2022-08-26 15:41:38 +00:00 |
|
Adrian Wälchli
|
eb233ea12d
|
Snapshot selected globals and restore them in spawned process (#13921)
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
|
2022-08-01 22:21:46 +00:00 |
|
Adrian Wälchli
|
81f149e9d4
|
Rename spawn-based launchers (#13743)
|
2022-07-23 11:48:15 -04:00 |
|
Adrian Wälchli
|
c3299d2c59
|
Add support for DDP fork (#13405)
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
|
2022-07-22 16:05:35 +00:00 |