Add register_all_kernels API to override PyTorch aten ops with oink CuTeDSL kernels#130

Merged

Laurawly merged 10 commits intomainfrom

register-all-kernels

Apr 22, 2026

Commits on Apr 9, 2026

Add register_all_kernels API and aten override for all oink kernels
jiannanWang
committed

Commits on Apr 13, 2026

Add vLLM-style stride guards and switch tests to pytest
jiannanWang
committed
Refine aten overrides: add env-gated tracing, remove debug prints
jiannanWang
committed
Refine aten overrides: add env-gated tracing, remove debug prints
jiannanWang
committed

Commits on Apr 14, 2026

Commits on Apr 21, 2026

Add register_all_kernels API and aten RMSNorm override with benchmarks
jiannanWang
committed

Commits on Apr 22, 2026