If you want to do PE post MBA, regardless of where you want to find work after bschool, go with the school with the best reputation and curriculum.
Berkley and UT-Austin are great bschools but Wharton is a better choice for PE. In PE, contacts and networks matter, especially your educational background. Your Wharton connection will take you far in this field.
Also, in the future, for any reason, you need to relocate. Wharton MBA will help you find opportunities easier (especially in PE) anywhere in US, or in other parts of the world for that matter.
This is not an assessment ranking one school against another. For PE, Wharton should be your best option and best bang for your money.