1; RUN: opt -S -mtriple=amdgcn-amd-amdhsa -passes=infer-address-spaces %s | FileCheck %s 2 3; CHECK-LABEL: @icmp_flat_cmp_self( 4; CHECK: %cmp = icmp eq ptr addrspace(3) %group.ptr.0, %group.ptr.0 5define i1 @icmp_flat_cmp_self(ptr addrspace(3) %group.ptr.0) #0 { 6 %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 7 %cmp = icmp eq ptr %cast0, %cast0 8 ret i1 %cmp 9} 10 11; CHECK-LABEL: @icmp_flat_flat_from_group( 12; CHECK: %cmp = icmp eq ptr addrspace(3) %group.ptr.0, %group.ptr.1 13define i1 @icmp_flat_flat_from_group(ptr addrspace(3) %group.ptr.0, ptr addrspace(3) %group.ptr.1) #0 { 14 %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 15 %cast1 = addrspacecast ptr addrspace(3) %group.ptr.1 to ptr 16 %cmp = icmp eq ptr %cast0, %cast1 17 ret i1 %cmp 18} 19 20; CHECK-LABEL: @icmp_mismatch_flat_from_group_private( 21; CHECK: %cast0 = addrspacecast ptr addrspace(5) %private.ptr.0 to ptr 22; CHECK: %cast1 = addrspacecast ptr addrspace(3) %group.ptr.1 to ptr 23; CHECK: %cmp = icmp eq ptr %cast0, %cast1 24define i1 @icmp_mismatch_flat_from_group_private(ptr addrspace(5) %private.ptr.0, ptr addrspace(3) %group.ptr.1) #0 { 25 %cast0 = addrspacecast ptr addrspace(5) %private.ptr.0 to ptr 26 %cast1 = addrspacecast ptr addrspace(3) %group.ptr.1 to ptr 27 %cmp = icmp eq ptr %cast0, %cast1 28 ret i1 %cmp 29} 30 31; CHECK-LABEL: @icmp_flat_group_flat( 32; CHECK: %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 33; CHECK: %cmp = icmp eq ptr %cast0, %flat.ptr.1 34define i1 @icmp_flat_group_flat(ptr addrspace(3) %group.ptr.0, ptr %flat.ptr.1) #0 { 35 %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 36 %cmp = icmp eq ptr %cast0, %flat.ptr.1 37 ret i1 %cmp 38} 39 40; CHECK-LABEL: @icmp_flat_flat_group( 41; CHECK: %cast1 = addrspacecast ptr addrspace(3) %group.ptr.1 to ptr 42; CHECK: %cmp = icmp eq ptr %flat.ptr.0, %cast1 43define i1 @icmp_flat_flat_group(ptr %flat.ptr.0, ptr addrspace(3) %group.ptr.1) #0 { 44 %cast1 = addrspacecast ptr addrspace(3) %group.ptr.1 to ptr 45 %cmp = icmp eq ptr %flat.ptr.0, %cast1 46 ret i1 %cmp 47} 48 49; Keeping as cmp addrspace(3)* is better 50; CHECK-LABEL: @icmp_flat_to_group_cmp( 51; CHECK: %cast0 = addrspacecast ptr %flat.ptr.0 to ptr addrspace(3) 52; CHECK: %cast1 = addrspacecast ptr %flat.ptr.1 to ptr addrspace(3) 53; CHECK: %cmp = icmp eq ptr addrspace(3) %cast0, %cast1 54define i1 @icmp_flat_to_group_cmp(ptr %flat.ptr.0, ptr %flat.ptr.1) #0 { 55 %cast0 = addrspacecast ptr %flat.ptr.0 to ptr addrspace(3) 56 %cast1 = addrspacecast ptr %flat.ptr.1 to ptr addrspace(3) 57 %cmp = icmp eq ptr addrspace(3) %cast0, %cast1 58 ret i1 %cmp 59} 60 61; FIXME: Should be able to ask target about how to constant fold the 62; constant cast if this is OK to change if 0 is a valid pointer. 63 64; CHECK-LABEL: @icmp_group_flat_cmp_null( 65; CHECK: %cmp = icmp eq ptr addrspace(3) %group.ptr.0, addrspacecast (ptr null to ptr addrspace(3)) 66define i1 @icmp_group_flat_cmp_null(ptr addrspace(3) %group.ptr.0) #0 { 67 %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 68 %cmp = icmp eq ptr %cast0, null 69 ret i1 %cmp 70} 71 72; CHECK-LABEL: @icmp_group_flat_cmp_constant_inttoptr( 73; CHECK: %cmp = icmp eq ptr addrspace(3) %group.ptr.0, addrspacecast (ptr inttoptr (i64 400 to ptr) to ptr addrspace(3)) 74define i1 @icmp_group_flat_cmp_constant_inttoptr(ptr addrspace(3) %group.ptr.0) #0 { 75 %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 76 %cmp = icmp eq ptr %cast0, inttoptr (i64 400 to ptr) 77 ret i1 %cmp 78} 79 80; CHECK-LABEL: @icmp_mismatch_flat_group_private_cmp_null( 81; CHECK: %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 82; CHECK: %cmp = icmp eq ptr %cast0, addrspacecast (ptr addrspace(5) null to ptr) 83define i1 @icmp_mismatch_flat_group_private_cmp_null(ptr addrspace(3) %group.ptr.0) #0 { 84 %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 85 %cmp = icmp eq ptr %cast0, addrspacecast (ptr addrspace(5) null to ptr) 86 ret i1 %cmp 87} 88 89; CHECK-LABEL: @icmp_mismatch_flat_group_private_cmp_undef( 90; CHECK: %cmp = icmp eq ptr addrspace(3) %group.ptr.0, undef 91define i1 @icmp_mismatch_flat_group_private_cmp_undef(ptr addrspace(3) %group.ptr.0) #0 { 92 %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 93 %cmp = icmp eq ptr %cast0, addrspacecast (ptr addrspace(5) undef to ptr) 94 ret i1 %cmp 95} 96 97@lds0 = internal addrspace(3) global i32 0, align 4 98@global0 = internal addrspace(1) global i32 0, align 4 99 100; CHECK-LABEL: @icmp_mismatch_flat_group_global_cmp_gv( 101; CHECK: %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 102; CHECK: %cmp = icmp eq ptr %cast0, addrspacecast (ptr addrspace(1) @global0 to ptr) 103define i1 @icmp_mismatch_flat_group_global_cmp_gv(ptr addrspace(3) %group.ptr.0) #0 { 104 %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 105 %cmp = icmp eq ptr %cast0, addrspacecast (ptr addrspace(1) @global0 to ptr) 106 ret i1 %cmp 107} 108 109; CHECK-LABEL: @icmp_mismatch_group_global_cmp_gv_gv( 110; CHECK: %cmp = icmp eq ptr addrspacecast (ptr addrspace(3) @lds0 to ptr), addrspacecast (ptr addrspace(1) @global0 to ptr) 111define i1 @icmp_mismatch_group_global_cmp_gv_gv(ptr addrspace(3) %group.ptr.0) #0 { 112 %cmp = icmp eq ptr addrspacecast (ptr addrspace(3) @lds0 to ptr), addrspacecast (ptr addrspace(1) @global0 to ptr) 113 ret i1 %cmp 114} 115 116; CHECK-LABEL: @icmp_group_flat_cmp_undef( 117; CHECK: %cmp = icmp eq ptr addrspace(3) %group.ptr.0, undef 118define i1 @icmp_group_flat_cmp_undef(ptr addrspace(3) %group.ptr.0) #0 { 119 %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 120 %cmp = icmp eq ptr %cast0, undef 121 ret i1 %cmp 122} 123 124; Test non-canonical orders 125; CHECK-LABEL: @icmp_mismatch_flat_group_private_cmp_null_swap( 126; CHECK: %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 127; CHECK: %cmp = icmp eq ptr addrspacecast (ptr addrspace(5) null to ptr), %cast0 128define i1 @icmp_mismatch_flat_group_private_cmp_null_swap(ptr addrspace(3) %group.ptr.0) #0 { 129 %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 130 %cmp = icmp eq ptr addrspacecast (ptr addrspace(5) null to ptr), %cast0 131 ret i1 %cmp 132} 133 134; CHECK-LABEL: @icmp_group_flat_cmp_undef_swap( 135; CHECK: %cmp = icmp eq ptr addrspace(3) undef, %group.ptr.0 136define i1 @icmp_group_flat_cmp_undef_swap(ptr addrspace(3) %group.ptr.0) #0 { 137 %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 138 %cmp = icmp eq ptr undef, %cast0 139 ret i1 %cmp 140} 141 142; CHECK-LABEL: @icmp_mismatch_flat_group_private_cmp_undef_swap( 143; CHECK: %cmp = icmp eq ptr addrspace(3) undef, %group.ptr.0 144define i1 @icmp_mismatch_flat_group_private_cmp_undef_swap(ptr addrspace(3) %group.ptr.0) #0 { 145 %cast0 = addrspacecast ptr addrspace(3) %group.ptr.0 to ptr 146 %cmp = icmp eq ptr addrspacecast (ptr addrspace(5) undef to ptr), %cast0 147 ret i1 %cmp 148} 149 150; CHECK-LABEL: @icmp_flat_flat_from_group_vector( 151; CHECK: %cmp = icmp eq <2 x ptr addrspace(3)> %group.ptr.0, %group.ptr.1 152define <2 x i1> @icmp_flat_flat_from_group_vector(<2 x ptr addrspace(3)> %group.ptr.0, <2 x ptr addrspace(3)> %group.ptr.1) #0 { 153 %cast0 = addrspacecast <2 x ptr addrspace(3)> %group.ptr.0 to <2 x ptr> 154 %cast1 = addrspacecast <2 x ptr addrspace(3)> %group.ptr.1 to <2 x ptr> 155 %cmp = icmp eq <2 x ptr> %cast0, %cast1 156 ret <2 x i1> %cmp 157} 158 159attributes #0 = { nounwind } 160