nir: calculate trip count for more loops