Split gpu 3body kernel into forces and rx