multi-gpu code now runs correctly for one gpu--is broken for more