diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2021-11-04 15:19:57 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-11-04 15:19:57 +0100 |
| commit | 0e431c203a66c6d48e6bb1efeda149460472a0f0 (patch) | |
| tree | 22a447a1dbb1505b18f9e426e1761cf6b328b6eb /test/conftest.py | |
| parent | 2fe69eb130827560ada704bc25998397c4357821 (diff) | |
[MRG] Add tests about type and GPU for emd/emd2 + 1d variants + wasserstein1d (#304)
* new test gpu
* pep 8 of couse
* debug torch
* jax with gpu
* device put
* device put
* it works
* emd1d and emd2_1d working
* emd_1d and emd2_1d done
* cleanup
* of course
* should work on gpu now
* tests done+ pep8
Diffstat (limited to 'test/conftest.py')
0 files changed, 0 insertions, 0 deletions