[Dev] Enable selective recompute for norm_out in GDN layers - #5443

[Dev] Enable selective recompute for norm_out in GDN layers#5443

wplf wants to merge 1 commit into

NVIDIA:devNVIDIA/Megatron-LM:devfrom

wplf:jinliangl/gdn-norm-out-recomputewplf/Megatron-LM:jinliangl/gdn-norm-out-recomputeCopy head branch name to clipboard