summaryrefslogtreecommitdiff
path: root/ot/gpu/__init__.py
blob: ed6dcc44eb164da5d993df04bb3ac0507da96827 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# -*- coding: utf-8 -*-

from . import bregman
from . import da
from .bregman import sinkhorn

# Author: Remi Flamary <remi.flamary@unice.fr>
#         Leo Gautheron <https://github.com/aje>
#
# License: MIT License

import warnings

warnings.warn("the ot.gpu module is deprecated because cudamat in no longer maintained", DeprecationWarning,
              stacklevel=2)

__all__ = ["bregman", "da", "sinkhorn"]