feat(gpu_prover): [WIP] Babybear NTT#194

Open

mcarilli wants to merge 37 commits intorr/av_gkr_compilerfrom

mc-babybear-ntt-rebase

Commits on Feb 2, 2026

Commits on Feb 3, 2026

wip
robik75
committed

Commits on Feb 4, 2026

compiles
mcarilli
committed

Commits on Feb 5, 2026

add device context with twiddles
mcarilli
committed
stage for L4 test
mcarilli
committed
stage for ncu source
mcarilli
committed
stage for full source profile
mcarilli
committed

Commits on Feb 6, 2026

try kernel with multiple blocks per SM
mcarilli
committed

Commits on Feb 10, 2026

coalesced experiment
mcarilli
committed
register pipeline experiment
mcarilli
committed
another experiment
mcarilli
committed
pipeline compute and LDGSTS
mcarilli
committed

Commits on Feb 11, 2026

producer-consumer kernel works
mcarilli
committed
fix ldgsts alignment
mcarilli
committed

Commits on Feb 13, 2026

use cmem for all warp-uniform twiddles
mcarilli
committed

Commits on Feb 16, 2026

logn=24 with split cmem and smem twiddles works
mcarilli
committed

Commits on Feb 17, 2026

logn=23 works
mcarilli
committed

Commits on Feb 20, 2026

first 16 stages of 3-pass work, also faster than sppark on L4 :)
mcarilli
committed
stage for 5090
mcarilli
committed
wip, debugging dataflow
mcarilli
committed

Commits on Feb 21, 2026

final stages experiments
mcarilli
committed

Commits on Feb 22, 2026

3 stage works, 7-8% faster than sppark on 5090
mcarilli
committed
first cut at logn=21,22,23 with same pattern
mcarilli
committed

Commits on Feb 23, 2026

three-pass works for logn=21,22,23,24, also three and two-pass support multiple columns
mcarilli
committed

Commits on Feb 24, 2026

wip: monomials to evals
mcarilli
committed

Commits on Feb 26, 2026

add option to write chunk-transposed monomials
mcarilli
committed

Commits on Feb 27, 2026

wip: monomials to evals
mcarilli
committed

Commits on Feb 28, 2026

remove experiment graveyard. first b2n kernel works.
mcarilli
committed

Commits on Mar 1, 2026

fix silly bugs. 3 pass monomials to evals works end to end
mcarilli
committed

Commits on Mar 2, 2026

fix output chaining in 3-pass, rustfmt
mcarilli
committed
wip: 2-pass monomials to evals
mcarilli
committed

Commits on Mar 3, 2026

2-pass monomials to evals dataflow works
mcarilli
committed
2-pass monomials to evals works for logn=23,24
mcarilli
committed
hacks
mcarilli
committed