[AArch64] NFC: Simplify the smstart/smstop pseudo. (#85067)

This is just a bit of cleanup to make the pseudo/code easier to
understand. This is based on the observation that we only need to pass
in a runtime value for 'pstate' if is actually needed for generating a
runtime check.