Cherry-pick https://github.com/NVIDIA/TensorRT-LLM/pull/5947#5989

lfr-0531 merged 1 commit intoNVIDIA:mainNVIDIA/TensorRT-LLM:mainfrom

lfr-0531:user/fanrongl/cherry_pick_illegal_mem_fixlfr-0531/TensorRT-LLM:user/fanrongl/cherry_pick_illegal_mem_fixCopy head branch name to clipboard