Fix ZeRO++ secondary shard copy for small params (#8210)