Numerical method for partial equilibrium flow