The role of spin-orbit coupling in the electronic structure of large spin Hall material IrO$_2$