Move the PME GPU clearing to the end of the step