Fix trtllm-gen attention illegal memory access#2002

yzh119 merged 1 commit intoflashinfer-ai:mainflashinfer-ai/flashinfer:mainfrom

Tom-Zheng:fix-trtllm-gen-attention-wrapperTom-Zheng/flashinfer:fix-trtllm-gen-attention-wrapperCopy head branch name to clipboard