Test-time scaling of diffusions with flow maps