blob: 12db60504e2b5721677b597cbf9f9b1295dca29c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
# -*- coding: utf-8 -*-
"""
GPU implementation for several OT solvers and utility
functions.
The GPU backend in handled by `cupy
<https://cupy.chainer.org/>`_.
.. warning::
This module is now deprecated and will be removed in future releases. POT
now privides a backend mechanism that allows for solving prolem on GPU wth
the pytorch backend.
.. warning::
Note that by default the module is not imported in :mod:`ot`. In order to
use it you need to explicitely import :mod:`ot.gpu` .
By default, the functions in this module accept and return numpy arrays
in order to proide drop-in replacement for the other POT function but
the transfer between CPU en GPU comes with a significant overhead.
In order to get the best performances, we recommend to give only cupy
arrays to the functions and desactivate the conversion to numpy of the
result of the function with parameter ``to_numpy=False``.
"""
# Author: Remi Flamary <remi.flamary@unice.fr>
# Leo Gautheron <https://github.com/aje>
#
# License: MIT License
import warnings
from . import bregman
from . import da
from .bregman import sinkhorn
from .da import sinkhorn_lpl1_mm
from . import utils
from .utils import dist, to_gpu, to_np
warnings.warn('This module is deprecated and will be removed in the next minor release of POT', category=DeprecationWarning)
__all__ = ["utils", "dist", "sinkhorn",
"sinkhorn_lpl1_mm", 'bregman', 'da', 'to_gpu', 'to_np']
|
