That system is based on security risk, not the actual offence per se. There is no extra punitive value attached to the offence in terms of prison conditions for the pure reason of the offence.  The reason why those inmates are kept separate is for their own safety (capital punishment should not be meted out by other prisoners!), not based on any ideal of punishment.