|
| 1 | +; NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py UTC_ARGS: --version 4 |
| 2 | +; RUN: llc -mattr=+sme -stop-after=finalize-isel < %s | FileCheck %s --check-prefix=CHECK-COALESCER-BARRIER |
| 3 | +; RUN: llc -mattr=+sme -stop-after=virtregrewriter < %s | FileCheck %s --check-prefix=CHECK-REGALLOC |
| 4 | + |
| 5 | +target triple = "aarch64" |
| 6 | + |
| 7 | +define void @dont_coalesce_args(<2 x i64> %a) "aarch64_pstate_sm_body" nounwind { |
| 8 | + ; CHECK-COALESCER-BARRIER-LABEL: name: dont_coalesce_args |
| 9 | + ; CHECK-COALESCER-BARRIER: bb.0 (%ir-block.0): |
| 10 | + ; CHECK-COALESCER-BARRIER-NEXT: liveins: $q0 |
| 11 | + ; CHECK-COALESCER-BARRIER-NEXT: {{ $}} |
| 12 | + ; CHECK-COALESCER-BARRIER-NEXT: [[COPY:%[0-9]+]]:fpr128 = COPY $q0 |
| 13 | + ; CHECK-COALESCER-BARRIER-NEXT: [[COALESCER_BARRIER_FPR128_:%[0-9]+]]:fpr128 = COALESCER_BARRIER_FPR128 [[COPY]] |
| 14 | + ; CHECK-COALESCER-BARRIER-NEXT: MSRpstatesvcrImm1 1, 1, csr_aarch64_smstartstop, implicit-def dead $nzcv, implicit $vg, implicit-def $vg |
| 15 | + ; CHECK-COALESCER-BARRIER-NEXT: [[DEF:%[0-9]+]]:zpr = IMPLICIT_DEF |
| 16 | + ; CHECK-COALESCER-BARRIER-NEXT: [[INSERT_SUBREG:%[0-9]+]]:zpr = INSERT_SUBREG [[DEF]], [[COALESCER_BARRIER_FPR128_]], %subreg.zsub |
| 17 | + ; CHECK-COALESCER-BARRIER-NEXT: ADJCALLSTACKDOWN 0, 0, implicit-def dead $sp, implicit $sp |
| 18 | + ; CHECK-COALESCER-BARRIER-NEXT: $z0 = COPY [[INSERT_SUBREG]] |
| 19 | + ; CHECK-COALESCER-BARRIER-NEXT: BL @scalable_args, csr_aarch64_sve_aapcs, implicit-def dead $lr, implicit $sp, implicit $z0, implicit-def $sp |
| 20 | + ; CHECK-COALESCER-BARRIER-NEXT: ADJCALLSTACKUP 0, 0, implicit-def dead $sp, implicit $sp |
| 21 | + ; CHECK-COALESCER-BARRIER-NEXT: MSRpstatesvcrImm1 1, 0, csr_aarch64_smstartstop, implicit-def dead $nzcv, implicit $vg, implicit-def $vg |
| 22 | + ; CHECK-COALESCER-BARRIER-NEXT: RET_ReallyLR |
| 23 | + ; |
| 24 | + ; CHECK-REGALLOC-LABEL: name: dont_coalesce_args |
| 25 | + ; CHECK-REGALLOC: bb.0 (%ir-block.0): |
| 26 | + ; CHECK-REGALLOC-NEXT: liveins: $q0 |
| 27 | + ; CHECK-REGALLOC-NEXT: {{ $}} |
| 28 | + ; CHECK-REGALLOC-NEXT: renamable $q0 = COALESCER_BARRIER_FPR128 killed renamable $q0 |
| 29 | + ; CHECK-REGALLOC-NEXT: STRQui killed renamable $q0, %stack.0, 0 :: (store (s128) into %stack.0) |
| 30 | + ; CHECK-REGALLOC-NEXT: MSRpstatesvcrImm1 1, 1, csr_aarch64_smstartstop, implicit-def dead $nzcv, implicit $vg, implicit-def $vg |
| 31 | + ; CHECK-REGALLOC-NEXT: renamable $q0 = LDRQui %stack.0, 0 :: (load (s128) from %stack.0) |
| 32 | + ; CHECK-REGALLOC-NEXT: renamable $q0 = KILL killed renamable $q0, implicit-def $z0 |
| 33 | + ; CHECK-REGALLOC-NEXT: ADJCALLSTACKDOWN 0, 0, implicit-def dead $sp, implicit $sp |
| 34 | + ; CHECK-REGALLOC-NEXT: BL @scalable_args, csr_aarch64_sve_aapcs, implicit-def dead $lr, implicit $sp, implicit $z0, implicit-def $sp |
| 35 | + ; CHECK-REGALLOC-NEXT: ADJCALLSTACKUP 0, 0, implicit-def dead $sp, implicit $sp |
| 36 | + ; CHECK-REGALLOC-NEXT: MSRpstatesvcrImm1 1, 0, csr_aarch64_smstartstop, implicit-def dead $nzcv, implicit $vg, implicit-def $vg |
| 37 | + ; CHECK-REGALLOC-NEXT: RET_ReallyLR |
| 38 | + %sa = call <vscale x 2 x i64> @llvm.vector.insert.nxv2i64.v2i64(<vscale x 2 x i64> poison, <2 x i64> %a, i64 0) |
| 39 | + call void @scalable_args(<vscale x 2 x i64> %sa) |
| 40 | + ret void |
| 41 | +} |
| 42 | + |
| 43 | +define <2 x i64> @dont_coalesce_res() "aarch64_pstate_sm_body" nounwind { |
| 44 | + ; CHECK-COALESCER-BARRIER-LABEL: name: dont_coalesce_res |
| 45 | + ; CHECK-COALESCER-BARRIER: bb.0 (%ir-block.0): |
| 46 | + ; CHECK-COALESCER-BARRIER-NEXT: MSRpstatesvcrImm1 1, 1, csr_aarch64_smstartstop, implicit-def dead $nzcv, implicit $vg, implicit-def $vg |
| 47 | + ; CHECK-COALESCER-BARRIER-NEXT: ADJCALLSTACKDOWN 0, 0, implicit-def dead $sp, implicit $sp |
| 48 | + ; CHECK-COALESCER-BARRIER-NEXT: BL @scalable_res, csr_aarch64_sve_aapcs, implicit-def dead $lr, implicit $sp, implicit-def $sp, implicit-def $z0 |
| 49 | + ; CHECK-COALESCER-BARRIER-NEXT: ADJCALLSTACKUP 0, 0, implicit-def dead $sp, implicit $sp |
| 50 | + ; CHECK-COALESCER-BARRIER-NEXT: [[COPY:%[0-9]+]]:zpr = COPY $z0 |
| 51 | + ; CHECK-COALESCER-BARRIER-NEXT: [[COPY1:%[0-9]+]]:fpr128 = COPY [[COPY]].zsub |
| 52 | + ; CHECK-COALESCER-BARRIER-NEXT: [[COALESCER_BARRIER_FPR128_:%[0-9]+]]:fpr128 = COALESCER_BARRIER_FPR128 [[COPY1]] |
| 53 | + ; CHECK-COALESCER-BARRIER-NEXT: MSRpstatesvcrImm1 1, 0, csr_aarch64_smstartstop, implicit-def dead $nzcv, implicit-def $q0, implicit $vg, implicit-def $vg |
| 54 | + ; CHECK-COALESCER-BARRIER-NEXT: $q0 = COPY [[COALESCER_BARRIER_FPR128_]] |
| 55 | + ; CHECK-COALESCER-BARRIER-NEXT: RET_ReallyLR implicit $q0 |
| 56 | + ; |
| 57 | + ; CHECK-REGALLOC-LABEL: name: dont_coalesce_res |
| 58 | + ; CHECK-REGALLOC: bb.0 (%ir-block.0): |
| 59 | + ; CHECK-REGALLOC-NEXT: MSRpstatesvcrImm1 1, 1, csr_aarch64_smstartstop, implicit-def dead $nzcv, implicit $vg, implicit-def $vg |
| 60 | + ; CHECK-REGALLOC-NEXT: ADJCALLSTACKDOWN 0, 0, implicit-def dead $sp, implicit $sp |
| 61 | + ; CHECK-REGALLOC-NEXT: BL @scalable_res, csr_aarch64_sve_aapcs, implicit-def dead $lr, implicit $sp, implicit-def $sp, implicit-def $z0 |
| 62 | + ; CHECK-REGALLOC-NEXT: ADJCALLSTACKUP 0, 0, implicit-def dead $sp, implicit $sp |
| 63 | + ; CHECK-REGALLOC-NEXT: renamable $q0 = KILL renamable $q0, implicit killed $z0 |
| 64 | + ; CHECK-REGALLOC-NEXT: renamable $q0 = COALESCER_BARRIER_FPR128 killed renamable $q0 |
| 65 | + ; CHECK-REGALLOC-NEXT: STRQui killed renamable $q0, %stack.0, 0 :: (store (s128) into %stack.0) |
| 66 | + ; CHECK-REGALLOC-NEXT: MSRpstatesvcrImm1 1, 0, csr_aarch64_smstartstop, implicit-def dead $nzcv, implicit-def dead $q0, implicit $vg, implicit-def $vg |
| 67 | + ; CHECK-REGALLOC-NEXT: $q0 = LDRQui %stack.0, 0 :: (load (s128) from %stack.0) |
| 68 | + ; CHECK-REGALLOC-NEXT: RET_ReallyLR implicit $q0 |
| 69 | + %sa = call <vscale x 2 x i64> @scalable_res() |
| 70 | + %res = call <2 x i64> @llvm.vector.extract.v2i64.nxv2i64(<vscale x 2 x i64> %sa, i64 0) |
| 71 | + ret <2 x i64> %res |
| 72 | +} |
| 73 | + |
| 74 | +define <2 x i64> @dont_coalesce_arg_that_is_also_res(<2 x i64> %a) "aarch64_pstate_sm_body" nounwind { |
| 75 | + ; CHECK-COALESCER-BARRIER-LABEL: name: dont_coalesce_arg_that_is_also_res |
| 76 | + ; CHECK-COALESCER-BARRIER: bb.0 (%ir-block.0): |
| 77 | + ; CHECK-COALESCER-BARRIER-NEXT: liveins: $q0 |
| 78 | + ; CHECK-COALESCER-BARRIER-NEXT: {{ $}} |
| 79 | + ; CHECK-COALESCER-BARRIER-NEXT: [[COPY:%[0-9]+]]:fpr128 = COPY $q0 |
| 80 | + ; CHECK-COALESCER-BARRIER-NEXT: [[COALESCER_BARRIER_FPR128_:%[0-9]+]]:fpr128 = COALESCER_BARRIER_FPR128 [[COPY]] |
| 81 | + ; CHECK-COALESCER-BARRIER-NEXT: MSRpstatesvcrImm1 1, 1, csr_aarch64_smstartstop, implicit-def dead $nzcv, implicit $vg, implicit-def $vg |
| 82 | + ; CHECK-COALESCER-BARRIER-NEXT: [[DEF:%[0-9]+]]:zpr = IMPLICIT_DEF |
| 83 | + ; CHECK-COALESCER-BARRIER-NEXT: [[INSERT_SUBREG:%[0-9]+]]:zpr = INSERT_SUBREG [[DEF]], [[COALESCER_BARRIER_FPR128_]], %subreg.zsub |
| 84 | + ; CHECK-COALESCER-BARRIER-NEXT: ADJCALLSTACKDOWN 0, 0, implicit-def dead $sp, implicit $sp |
| 85 | + ; CHECK-COALESCER-BARRIER-NEXT: $z0 = COPY [[INSERT_SUBREG]] |
| 86 | + ; CHECK-COALESCER-BARRIER-NEXT: BL @scalable_args, csr_aarch64_sve_aapcs, implicit-def dead $lr, implicit $sp, implicit $z0, implicit-def $sp |
| 87 | + ; CHECK-COALESCER-BARRIER-NEXT: ADJCALLSTACKUP 0, 0, implicit-def dead $sp, implicit $sp |
| 88 | + ; CHECK-COALESCER-BARRIER-NEXT: [[COALESCER_BARRIER_FPR128_1:%[0-9]+]]:fpr128 = COALESCER_BARRIER_FPR128 [[COALESCER_BARRIER_FPR128_]] |
| 89 | + ; CHECK-COALESCER-BARRIER-NEXT: MSRpstatesvcrImm1 1, 0, csr_aarch64_smstartstop, implicit-def dead $nzcv, implicit-def $q0, implicit $vg, implicit-def $vg |
| 90 | + ; CHECK-COALESCER-BARRIER-NEXT: $q0 = COPY [[COALESCER_BARRIER_FPR128_1]] |
| 91 | + ; CHECK-COALESCER-BARRIER-NEXT: RET_ReallyLR implicit $q0 |
| 92 | + ; |
| 93 | + ; CHECK-REGALLOC-LABEL: name: dont_coalesce_arg_that_is_also_res |
| 94 | + ; CHECK-REGALLOC: bb.0 (%ir-block.0): |
| 95 | + ; CHECK-REGALLOC-NEXT: liveins: $q0 |
| 96 | + ; CHECK-REGALLOC-NEXT: {{ $}} |
| 97 | + ; CHECK-REGALLOC-NEXT: renamable $q0 = COALESCER_BARRIER_FPR128 killed renamable $q0 |
| 98 | + ; CHECK-REGALLOC-NEXT: STRQui killed renamable $q0, %stack.0, 0 :: (store (s128) into %stack.0) |
| 99 | + ; CHECK-REGALLOC-NEXT: MSRpstatesvcrImm1 1, 1, csr_aarch64_smstartstop, implicit-def dead $nzcv, implicit $vg, implicit-def $vg |
| 100 | + ; CHECK-REGALLOC-NEXT: renamable $q0 = LDRQui %stack.0, 0 :: (load (s128) from %stack.0) |
| 101 | + ; CHECK-REGALLOC-NEXT: renamable $q0 = KILL killed renamable $q0, implicit-def $z0 |
| 102 | + ; CHECK-REGALLOC-NEXT: ADJCALLSTACKDOWN 0, 0, implicit-def dead $sp, implicit $sp |
| 103 | + ; CHECK-REGALLOC-NEXT: BL @scalable_args, csr_aarch64_sve_aapcs, implicit-def dead $lr, implicit $sp, implicit $z0, implicit-def $sp |
| 104 | + ; CHECK-REGALLOC-NEXT: ADJCALLSTACKUP 0, 0, implicit-def dead $sp, implicit $sp |
| 105 | + ; CHECK-REGALLOC-NEXT: renamable $q0 = LDRQui %stack.0, 0 :: (load (s128) from %stack.0) |
| 106 | + ; CHECK-REGALLOC-NEXT: renamable $q0 = COALESCER_BARRIER_FPR128 killed renamable $q0 |
| 107 | + ; CHECK-REGALLOC-NEXT: STRQui killed renamable $q0, %stack.0, 0 :: (store (s128) into %stack.0) |
| 108 | + ; CHECK-REGALLOC-NEXT: MSRpstatesvcrImm1 1, 0, csr_aarch64_smstartstop, implicit-def dead $nzcv, implicit-def dead $q0, implicit $vg, implicit-def $vg |
| 109 | + ; CHECK-REGALLOC-NEXT: $q0 = LDRQui %stack.0, 0 :: (load (s128) from %stack.0) |
| 110 | + ; CHECK-REGALLOC-NEXT: RET_ReallyLR implicit $q0 |
| 111 | + %sa = call <vscale x 2 x i64> @llvm.vector.insert.nxv2i64.v2i64(<vscale x 2 x i64> poison, <2 x i64> %a, i64 0) |
| 112 | + call void @scalable_args(<vscale x 2 x i64> %sa) |
| 113 | + ret <2 x i64> %a |
| 114 | +} |
| 115 | + |
| 116 | +declare void @scalable_args(<vscale x 2 x i64>) "aarch64_pstate_sm_enabled" |
| 117 | +declare <vscale x 2 x i64> @llvm.vector.insert.nxv2i64.v2i64(<vscale x 2 x i64>, <2 x i64>, i64) |
| 118 | + |
| 119 | +declare <vscale x 2 x i64> @scalable_res() "aarch64_pstate_sm_enabled" |
| 120 | +declare <2 x i64> @llvm.vector.extract.v2i64.nxv2i64(<vscale x 2 x i64>, i64) |
0 commit comments