[GDN] Keep A_log and out_norm in FP32 for numerical stability#3634

cuichenx wants to merge 1 commit intoNVIDIA:mainNVIDIA/Megatron-LM:mainfrom

cuichenx:chcui/qwen3.5_precisioncuichenx/Megatron-LM:chcui/qwen3.5_precisionCopy head branch name to clipboard