[MRG] Avoid changing precision in the backend by kachayev · Pull Request #572 · PythonOT/POT
Types of changes
Fix previously added (see PR #507) precision change when computing norm on the PyTorch backend.
Motivation and context / Related issue
Precision change is not necessary, and fails on devices which don't have support for double precision.
Related discussion #570.
How has this been tested (if it applies)
n/a
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.