Skip to content

Commit b29a7e1

Browse files
whjthuvoltjia
andauthored
Add SSA compiler pipeline and unified backends (#164)
* Add SSA compiler pipeline and unified backends * refine ssa multibackend PR boundaries and pass registration * move backend ssa pass registration into backend modules * Add backend SSA pass contracts and registry validation * revert test_aot_auto_tuning.py * fix ruff * style: satisfy contributing style checks after rebase * Refactor SSA compiler naming (#169) * Refactor IR dataclass constructors (#172) * unify SSA compilation and executable multi-backend support * Fix ruff error, refine SSA compiler structure and address review feedback * Refactor SSA backend pipeline and restore Triton runtime autotining * Removing TVM backend and refine Triton, CUDA, TileLang backends * Fix zero-like GPU launches, architecture-aware caching, and JIT API * Fix Triton AOT multi-context launches (#195) * Add Triton AOT multi-context design * Fix Triton AOT multi-context launches * Cover fresh Triton AOT context guard * Remove development process documents * Address remaining SSA review comments (#196) * Address remaining SSA review comments * Sort test imports for Ruff * Remove redundant entrypoint assertions --------- Co-authored-by: Jiacheng Huang <45955067+voltjia@users.noreply.github.com>
1 parent c9ebd49 commit b29a7e1

67 files changed

Lines changed: 18573 additions & 3127 deletions

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

src/ninetoothed/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from ninetoothed.build import build
2+
from ninetoothed.compiler.jit import jit
23
from ninetoothed.dtype import (
34
bfloat16,
45
float16,
@@ -15,7 +16,6 @@
1516
)
1617
from ninetoothed.eval import _eval as eval
1718
from ninetoothed.eval import _subs as subs
18-
from ninetoothed.jit import jit
1919
from ninetoothed.make import make
2020
from ninetoothed.symbol import Symbol, block_size
2121
from ninetoothed.tensor import Tensor

0 commit comments

Comments
 (0)