MULTIGPU: fixed the computation bug