[PATCH v4 10/21] arm64/sme: Implement context switching for ZT0