It depends. The Pit Fiend has greater action economy but both the balor and the pit fiend are immune to fire and poison. In a straight up fight the the pit fiend still has more attacks and thus does more damage. But with that being said, what you could do with the balor is have it drop the flame...