kokkos: added some more code to kokkos fix langevin