Add simplify_pw_aff to generate simpler loop bounds