I run a very similar setup, with an AX1200i PSU. Including going from a 1080 to a watercooled 3090.
I doubt it's heat, the system would tend to freeze instead of rebooting.
I'm more inclined to suspect the PSU is cutting out.
Does your 3080 power cables all come separately from the PSU, or are they on any kind of splitter or shared cable?