llvm/test/CodeGen/AMDGPU/preserve-only-inactive-lane.mir

   1 # NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
   2 # RUN: llc -mtriple=amdgcn-amd-amdhsa -mcpu=gfx906 -verify-machineinstrs -run-pass=prologepilog -o - %s | FileCheck -check-prefix=GCN %s
   3
   4 # The scratch register vgpr0 is used here to spill sgpr35 register.
   5 # The function also returns a value that goes via. vgpr0.
   6 # This test is to ensure the vgpr0 store/restore inserted at the prolog/epilog
   7 # preserves only its inactive lanes to ensure the outgoing value will not be overwritten.
   8
   9 ---
  10 name: preserve_scratch_vgpr_inactive_lanes
  11 tracksRegLiveness: true
  12 machineFunctionInfo:
  13   scratchRSrcReg:  $sgpr0_sgpr1_sgpr2_sgpr3
  14   frameOffsetReg:  $sgpr33
  15   stackPtrOffsetReg:  $sgpr32
  16
  17 body:             |
  18   bb.0:
  19     liveins: $sgpr35, $vgpr0, $sgpr30_sgpr31
  20     ; GCN-LABEL: name: preserve_scratch_vgpr_inactive_lanes
  21     ; GCN: liveins: $sgpr35, $vgpr0, $sgpr30_sgpr31
  22     ; GCN-NEXT: {{  $}}
  23     ; GCN-NEXT: $sgpr4_sgpr5 = S_XOR_SAVEEXEC_B64 -1, implicit-def $exec, implicit-def dead $scc, implicit $exec
  24     ; GCN-NEXT: BUFFER_STORE_DWORD_OFFSET $vgpr0, $sgpr0_sgpr1_sgpr2_sgpr3, $sgpr32, 0, 0, 0, implicit $exec :: (store (s32) into %stack.0, addrspace 5)
  25     ; GCN-NEXT: $exec = S_MOV_B64 killed $sgpr4_sgpr5
  26     ; GCN-NEXT: renamable $vgpr0 = SI_SPILL_S32_TO_VGPR $sgpr35, 0, killed $vgpr0
  27     ; GCN-NEXT: $sgpr35 = S_MOV_B32 5
  28     ; GCN-NEXT: $sgpr35 = SI_RESTORE_S32_FROM_VGPR $vgpr0, 0
  29     ; GCN-NEXT: renamable $vgpr0 = V_MOV_B32_e32 10, implicit $exec
  30     ; GCN-NEXT: $sgpr4_sgpr5 = S_XOR_SAVEEXEC_B64 -1, implicit-def $exec, implicit-def dead $scc, implicit $exec
  31     ; GCN-NEXT: $vgpr0 = BUFFER_LOAD_DWORD_OFFSET $sgpr0_sgpr1_sgpr2_sgpr3, $sgpr32, 0, 0, 0, implicit $exec, implicit $vgpr0(tied-def 0) :: (load (s32) from %stack.0, addrspace 5)
  32     ; GCN-NEXT: $exec = S_MOV_B64 killed $sgpr4_sgpr5
  33     ; GCN-NEXT: S_SETPC_B64_return killed renamable $sgpr30_sgpr31, implicit $vgpr0
  34     renamable $vgpr0 = SI_SPILL_S32_TO_VGPR $sgpr35, 0, killed $vgpr0
  35     $sgpr35 = S_MOV_B32 5
  36     $sgpr35 = SI_RESTORE_S32_FROM_VGPR $vgpr0, 0
  37     renamable $vgpr0 = V_MOV_B32_e32 10, implicit $exec
  38     S_SETPC_B64_return killed renamable $sgpr30_sgpr31, implicit $vgpr0
  39 ...