Fix visualization when data is from more than 1 gpu#938

Tcc0403 wants to merge 3 commits intomainlinkedin/Liger-Kernel:mainfrom

tcc/bench_vislinkedin/Liger-Kernel:tcc/bench_visCopy head branch name to clipboard