Yes, the SWCG should be AFTER the heater.
Chlorine levels immediately downstream of the SWCG can be high enough to cause damage to the heat pump over time. Perhaps if your unit has a titanium heat exchanger, this may not apply . . . but I'd call the manufacturer of YOUR unit and ask and see.
Bookmarks