; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sme -verify-machineinstrs < %s | FileCheck %s
; RUN: llc -mtriple=aarch64-apple-darwin -mattr=+sme -verify-machineinstrs < %s | FileCheck %s --check-prefix=DARWIN
; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sme -verify-machineinstrs -stop-after=finalize-isel < %s | FileCheck %s --check-prefix=CHECK-CSRMASK
; RUN: llc -mtriple=aarch64-apple-darwin -mattr=+sme -verify-machineinstrs -stop-after=finalize-isel < %s | FileCheck %s --check-prefix=CHECK-CSRMASK

; Test that the PCS attribute is accepted and uses the correct register mask.
;

define void @test_sme_calling_convention_x0() nounwind {
; CHECK-LABEL: test_sme_calling_convention_x0:
; CHECK:       // %bb.0:
; CHECK-NEXT:    str x30, [sp, #-16]! // 8-byte Folded Spill
; CHECK-NEXT:    bl __arm_tpidr2_save
; CHECK-NEXT:    ldr x30, [sp], #16 // 8-byte Folded Reload
; CHECK-NEXT:    ret
; DARWIN-LABEL: test_sme_calling_convention_x0:
; DARWIN:    stp	x29, x30, [sp, #-16]!
; DARWIN:    bl ___arm_tpidr2_save
; DARWIN:    ldp	x29, x30, [sp], #16
; DARWIN:    ret
;
; CHECK-CSRMASK-LABEL: name: test_sme_calling_convention_x0
; CHECK-CSRMASK: BL @__arm_tpidr2_save, csr_aarch64_sme_abi_support_routines_preservemost_from_x0
  call aarch64_sme_preservemost_from_x0 void @__arm_tpidr2_save()
  ret void
}

define i64 @test_sme_calling_convention_x1() nounwind {
; CHECK-LABEL: test_sme_calling_convention_x1:
; CHECK:       // %bb.0:
; CHECK-NEXT:    str x30, [sp, #-16]! // 8-byte Folded Spill
; CHECK-NEXT:    bl __arm_get_current_vg
; CHECK-NEXT:    ldr x30, [sp], #16 // 8-byte Folded Reload
; CHECK-NEXT:    ret
; DARWIN-LABEL: test_sme_calling_convention_x1:
; DARWIN:    stp        x29, x30, [sp, #-16]!
; DARWIN:    bl ___arm_get_current_vg
; DARWIN:    ldp        x29, x30, [sp], #16
; DARWIN:    ret
;
; CHECK-CSRMASK-LABEL: name: test_sme_calling_convention_x1
; CHECK-CSRMASK: BL @__arm_get_current_vg, csr_aarch64_sme_abi_support_routines_preservemost_from_x1
  %vg = call aarch64_sme_preservemost_from_x1 i64 @__arm_get_current_vg()
  ret i64 %vg
}

define i64 @test_sme_calling_convention_x2() nounwind {
; CHECK-LABEL: test_sme_calling_convention_x2:
; CHECK:       // %bb.0:
; CHECK-NEXT:    str x30, [sp, #-16]! // 8-byte Folded Spill
; CHECK-NEXT:    bl __arm_sme_state
; CHECK-NEXT:    ldr x30, [sp], #16 // 8-byte Folded Reload
; CHECK-NEXT:    ret
; DARWIN-LABEL: test_sme_calling_convention_x2:
; DARWIN:    stp	x29, x30, [sp, #-16]!
; DARWIN:    bl ___arm_sme_state
; DARWIN:    ldp	x29, x30, [sp], #16
; DARWIN:    ret
;
; CHECK-CSRMASK-LABEL: name: test_sme_calling_convention_x2
; CHECK-CSRMASK: BL @__arm_sme_state, csr_aarch64_sme_abi_support_routines_preservemost_from_x2
  %pstate = call aarch64_sme_preservemost_from_x2 {i64, i64} @__arm_sme_state()
  %pstate.sm = extractvalue {i64, i64} %pstate, 0
  ret i64 %pstate.sm
}

declare void @__arm_tpidr2_save()
declare i64 @__arm_get_current_vg()
declare {i64, i64} @__arm_sme_state()