; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; RUN: llc -mtriple=aarch64 -mattr=+sme -verify-machineinstrs < %s | FileCheck %s

define void @toggle_pstate_za() {
; CHECK-LABEL: toggle_pstate_za:
; CHECK:       // %bb.0:
; CHECK-NEXT:    smstart za
; CHECK-NEXT:    smstop za
; CHECK-NEXT:    ret
  call void @llvm.aarch64.sme.za.enable()
  call void @llvm.aarch64.sme.za.disable()
  ret void
}

declare void @llvm.aarch64.sme.za.enable()
declare void @llvm.aarch64.sme.za.disable()