[Bugfix] Do not write segment softmax state at NUM_SEGMENTS_PER_SEQ == 1 - #5935

[Bugfix] Do not write segment softmax state at NUM_SEGMENTS_PER_SEQ == 1#5935

k50112113 merged 1 commit into

ROCm:mainROCm/aiter:mainfrom

sogalin:fix-mla-decode-numsegments1sogalin/aiter:fix-mla-decode-numsegments1Copy head branch name to clipboard