Check new amen_sum with repmat instead of kron in the CP part