[Qwen3.5] Fix caching bug in GDN layer for autoregressive mode#3907

Rohan-Bierneni wants to merge 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

rbierneni-test-q3-cachingAI-Hypercomputer/maxtext:rbierneni-test-q3-cachingCopy head branch name to clipboard