Enable selective recompute for `norm_out` in GDN layers - #4715

Merged

xuantengh merged 7 commits into

NVIDIA:mainfrom

xuantengh:xuantengh/gdn_recompute

Jun 8, 2026

Commits on May 28, 2026

add norm_out recompute in gdn
xuantengh
committed
a6aefdf
add gdn test
xuantengh
committed
edfe19d
rebase upstream
xuantengh
committed
8552ddf
update test
xuantengh
committed
2806fec
rebase for thd layout
xuantengh
committed
3955c76
include a2a inside the recompute region
xuantengh
committed
89bffb1
add extra assertion
xuantengh
committed
e3d7d0d