revert changes in src/petuum_ps