I assumed a square one would apply heat evenly.
You're correct, a large round one would work too, but I'm uncertain if you mean to use the smaller circular that came with it or buy a large one that would fit around the square IC.
If you mean buy a large circular to fit around the square IC, then I'd imagine buying a square would be ideal.