DeepGEMM/deep_gemm/jit/__init__.py
Zihua Wu 27cd276e19 [wip] refactor: compile to .cubin
Signed-off-by: Zihua Wu <13583761+lucifer1004@users.noreply.github.com>
2025-04-22 08:08:40 +00:00

4 lines
107 B
Python

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