perf(gemma4): select decode down weights in output blocks - #487

Closed

AdnanHoque wants to merge 8 commits into

torch-spyre:mainfrom

AdnanHoque:perf/gemma4-decode-down-blocks

Commits on Sep 8, 2026

perf(gemma4): pair decode route scheduling with indexed-selection layouts
AdnanHoque
committed
b1accd5
refactor(gemma4): share the scoped decode call and cleanup path
AdnanHoque
committed
6cf881d
perf(gemma4): select decode down weights in output blocks
AdnanHoque
committed
7fddbd5
test(gemma4): guard every full down gather when blocks are enabled
AdnanHoque
committed
e462c2b
perf(gemma4): enable paired route scheduling automatically
AdnanHoque
committed
c04fa16
perf(gemma4): enable supported decode blocks and inherit automatic route scheduling
AdnanHoque
committed
4e05058
fix(gemma4): include checkpoint BF16 in automatic optimization selection
AdnanHoque
committed
ecdd49c
test(gemma4): reject mixed 16-bit optimization inputs
AdnanHoque
committed
d7de86d