Overhaul and stabilize AOT kernel launching mechanism#136

Merged

voltjia merged 8 commits intomasterfrom

overhaul-and-stabilize-aot-kernel-launching-mechanism

Mar 5, 2026

Commits on Mar 5, 2026

Move the `_run_launch_func` helper inside `_generate_launch_func` and have it return the wrapper function directly
voltjia
committed
Move AOT kernel launch utilities from `tests/test_aot.py` to `src/ninetoothed/aot.py`
voltjia
committed
Return launch function directly from `ninetoothed.aot` and `ninetoothed.build`
voltjia
committed
Update the kernel launching mechanism to accept `ninetoothed` data types
voltjia
committed
Check kernel launch return value in `_run_launch_func`
voltjia
committed
Keep reference to original `torch.Tensor` in `_ArgumentTensor`
voltjia
committed
Use current stream for kernel launch
voltjia
committed
Load library via temporary copy
voltjia
committed