Added cpu communication after restriction and prolongation