multiGPU infer hanging.

[yufang67]

i use 0.15.0.dev version. The model initialized by
llm=tensorrt_llm.LLM( path, tokenizer, dtype="float16", build_config=build_config, tensor_parallel_size=2, pipeline_parallel_size=1, )
Then, the infer run on a 4xa100 node with llm.generate(input, sampling_params).
The infer run through all samples and reach the end of the main program, but hang there.
Did i miss any config ?

Thanks