File size: 345 Bytes
008bd6b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | """
cuTENSOR Wrapper
Use `cupy_backends.cuda.libs.cutensor` directly in CuPy codebase.
"""
from __future__ import annotations
available = True
try:
from cupy_backends.cuda.libs.cutensor import * # NOQA
except ImportError as e:
available = False
from cupy._environment import _preload_warning
_preload_warning('cutensor', e)
|