If I compute the grad(U) once every time step and then use this sum operation for each z I believe it should be identical unless I'm missing something crucial.