summaryrefslogtreecommitdiffstats
path: root/llvm/test/Transforms/InferAddressSpaces/AMDGPU/address-space-id-funcs.ll
blob: 63cabb97c237feab6ad5a3ea57ccce4d26ba4c32 (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
47
48
49
50
51
52
53
54
55
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt -S -mtriple=amdgcn-amd-amdhsa -infer-address-spaces -instsimplify %s | FileCheck %s

define amdgpu_kernel void @is_local_true(i8 addrspace(3)* %lptr) {
; CHECK-LABEL: @is_local_true(
; CHECK-NEXT:    store i32 1, i32 addrspace(1)* undef
; CHECK-NEXT:    ret void
;
  %cast = addrspacecast i8 addrspace(3)* %lptr to i8*
  %is.shared = call i1 @llvm.amdgcn.is.shared(i8* %cast)
  %ext = zext i1 %is.shared to i32
  store i32 %ext, i32 addrspace(1)* undef
  ret void
}

define amdgpu_kernel void @is_local_false(i8 addrspace(1)* %gptr) {
; CHECK-LABEL: @is_local_false(
; CHECK-NEXT:    store i32 0, i32 addrspace(1)* undef
; CHECK-NEXT:    ret void
;
  %cast = addrspacecast i8 addrspace(1)* %gptr to i8*
  %is.shared = call i1 @llvm.amdgcn.is.shared(i8* %cast)
  %ext = zext i1 %is.shared to i32
  store i32 %ext, i32 addrspace(1)* undef
  ret void
}

define void @is_private_true(i8 addrspace(5)* %lptr) {
; CHECK-LABEL: @is_private_true(
; CHECK-NEXT:    store i32 1, i32 addrspace(1)* undef
; CHECK-NEXT:    ret void
;
  %cast = addrspacecast i8 addrspace(5)* %lptr to i8*
  %is.private = call i1 @llvm.amdgcn.is.private(i8* %cast)
  %ext = zext i1 %is.private to i32
  store i32 %ext, i32 addrspace(1)* undef
  ret void
}

define void @is_private_false(i8 addrspace(1)* %gptr) {
; CHECK-LABEL: @is_private_false(
; CHECK-NEXT:    store i32 0, i32 addrspace(1)* undef
; CHECK-NEXT:    ret void
;
  %cast = addrspacecast i8 addrspace(1)* %gptr to i8*
  %is.private = call i1 @llvm.amdgcn.is.private(i8* %cast)
  %ext = zext i1 %is.private to i32
  store i32 %ext, i32 addrspace(1)* undef
  ret void
}

declare i1 @llvm.amdgcn.is.shared(i8* nocapture) #0
declare i1 @llvm.amdgcn.is.private(i8* nocapture) #0

attributes #0 = { nounwind readnone speculatable }
OpenPOWER on IntegriCloud