nir: pass nir_op to calculate_iterations()