deep_gemm/jit/__init__.py (
3
lines of code) (
raw
):
from .compiler import get_nvcc_compiler, build from .template import cpp_format, generate from .runtime import Runtime