DeepGEMM/deep_gemm/jit/__init__.py
Zihua Wu 46762b6903 feat: make API more general
Signed-off-by: Zihua Wu <13583761+lucifer1004@users.noreply.github.com>
2025-04-23 02:34:23 -07:00

4 lines
136 B
Python

from .compiler import get_nvcc_compiler, build, NvccCompiler, NvrtcCompiler
from .template import generate
from .runtime import Runtime