53 lines
953 B
Python
53 lines
953 B
Python
"""isort:skip_file"""
|
|
__version__ = '2.0.0'
|
|
|
|
# ---------------------------------------
|
|
# Note: import order is significant here.
|
|
|
|
# TODO: torch needs to be imported first
|
|
# or pybind11 shows `munmap_chunk(): invalid pointer`
|
|
import torch # noqa: F401
|
|
|
|
# submodules
|
|
from . import impl
|
|
from .utils import (
|
|
cdiv,
|
|
MockTensor,
|
|
next_power_of_2,
|
|
reinterpret,
|
|
TensorWrapper,
|
|
)
|
|
from .runtime import (
|
|
autotune,
|
|
Config,
|
|
heuristics,
|
|
JITFunction,
|
|
KernelInterface,
|
|
)
|
|
from .runtime.jit import jit
|
|
from .compiler import compile, CompilationError
|
|
from . import language
|
|
from . import testing
|
|
from . import ops
|
|
|
|
__all__ = [
|
|
"autotune",
|
|
"cdiv",
|
|
"CompilationError",
|
|
"compile",
|
|
"Config",
|
|
"heuristics",
|
|
"impl",
|
|
"jit",
|
|
"JITFunction",
|
|
"KernelInterface",
|
|
"language",
|
|
"MockTensor",
|
|
"next_power_of_2",
|
|
"ops",
|
|
"reinterpret",
|
|
"runtime",
|
|
"TensorWrapper",
|
|
"testing",
|
|
]
|