gpu version working for non-corotational models