aboutsummaryrefslogtreecommitdiff
path: root/llvm/test/CodeGen/AMDGPU/scratch-pointer-sink.ll
blob: cdaac14833e0ef0e163f9c9d38815b8c60cac8fb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 3
; RUN: llc -mtriple=amdgcn -mcpu=gfx1100 -verify-machineinstrs < %s | FileCheck %s -check-prefixes=GCN
; RUN: llc -global-isel -mtriple=amdgcn -mcpu=gfx1100 -verify-machineinstrs < %s | FileCheck %s -check-prefixes=GISEL

define amdgpu_gfx i32 @sink_scratch_pointer(ptr addrspace(5) %stack, i32 inreg %flag) {
; GCN-LABEL: sink_scratch_pointer:
; GCN:       ; %bb.0:
; GCN-NEXT:    s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0)
; GCN-NEXT:    s_cmp_lg_u32 s4, 0
; GCN-NEXT:    s_cbranch_scc0 .LBB0_2
; GCN-NEXT:  ; %bb.1: ; %bb2
; GCN-NEXT:    scratch_load_b32 v0, v0, off offset:-4
; GCN-NEXT:    s_waitcnt vmcnt(0)
; GCN-NEXT:    s_setpc_b64 s[30:31]
; GCN-NEXT:  .LBB0_2: ; %bb1
; GCN-NEXT:    v_mov_b32_e32 v1, 1
; GCN-NEXT:    scratch_store_b32 v0, v1, off offset:-4
; GCN-NEXT:    v_mov_b32_e32 v0, 0
; GCN-NEXT:    s_setpc_b64 s[30:31]
;
; GISEL-LABEL: sink_scratch_pointer:
; GISEL:       ; %bb.0:
; GISEL-NEXT:    s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0)
; GISEL-NEXT:    s_cmp_lg_u32 s4, 0
; GISEL-NEXT:    s_cbranch_scc0 .LBB0_2
; GISEL-NEXT:  ; %bb.1: ; %bb2
; GISEL-NEXT:    scratch_load_b32 v0, v0, off offset:-4
; GISEL-NEXT:    s_waitcnt vmcnt(0)
; GISEL-NEXT:    s_setpc_b64 s[30:31]
; GISEL-NEXT:  .LBB0_2: ; %bb1
; GISEL-NEXT:    v_mov_b32_e32 v1, 1
; GISEL-NEXT:    scratch_store_b32 v0, v1, off offset:-4
; GISEL-NEXT:    v_mov_b32_e32 v0, 0
; GISEL-NEXT:    s_setpc_b64 s[30:31]
  %ptr = getelementptr inbounds i32, ptr addrspace(5) %stack, i32 -1
  %cond = icmp eq i32 %flag, 0
  br i1 %cond, label %bb1, label %bb2

bb1:
  store i32 1, ptr addrspace(5) %ptr, align 4
  ret i32 0

bb2:
  %value = load i32, ptr addrspace(5) %ptr, align 4
  ret i32 %value
}