Make the wait on nonbonded GPU results conditional