[PATCH v2 2/3] kselftest/arm64: Expand the set of ZA writes fp-ptrace does