[MRG] Solve example throwing an error when executed on a GPU by ncassereau · Pull Request #391 · PythonOT/POT
Types of changes
Changing code for backend example
Motivation and context / Related issue
Resolves #389
How has this been tested (if it applies)
Tested on Jean Zay with a V100
PR checklist
- I have read the CONTRIBUTING document.
- The documentation is up-to-date with the changes I made (check build artifacts).
- All tests passed, and additional code has been covered with new tests.
- I have added the PR and Issue fix to the RELEASES.md file.