nki(radix_sort): NKI (Trainium) implementation - #296

Open

bowencui123 wants to merge 8 commits into

mainfrom

bowen/nki/radix_sort

Commits on Aug 29, 2026

nki(radix_sort): NKI (Trainium) implementation

committed
1684cd5
nki(radix_sort): NkiAutotuner wiring (`block_size`)

bowencui123
and
claude
committed
11f19c5

Commits on Aug 30, 2026

review(nki): device-dispatch the torch baseline; refresh the stale timing note

bowencui123
and
claude
committed
fc756f4

Commits on Aug 31, 2026

style(nki): strip comments and docstrings per review

bowencui123
and
claude
committed
8b2d4c2

Commits on Sep 5, 2026

Merge branch 'main' into bowen/nki/radix_sort
Cecilia123li
authored
0e84833

Commits on Sep 23, 2026

Merge remote-tracking branch 'origin/main' into bowen/nki/radix_sort
bowencui123
committed
59af244
nki(radix_sort): migrate to the tilebench package layout

bowencui123
and
claude
committed
d52cf95

Commits on Sep 28, 2026

nki(radix_sort): benchmark warmup/repeat 1/3; drop comments

bowencui123
and
claude
committed
a27cf8e