1 ; RUN: llc -march=amdgcn -mcpu=tonga -mattr=-flat-for-global -verify-machineinstrs < %s | FileCheck -check-prefix=VI -check-prefix=GCN %s 2 ; RUN: llc -march=amdgcn -mcpu=bonaire -verify-machineinstrs < %s | FileCheck -check-prefix=CI -check-prefix=GCN %s 3 4 ; FIXME: Need to handle non-uniform case for function below (load without gep). 5 ; GCN-LABEL: {{^}}v_test_sub_i16: 6 ; VI: flat_load_ushort [[A:v[0-9]+]] 7 ; VI: flat_load_ushort [[B:v[0-9]+]] 8 ; VI: v_sub_u16_e32 [[ADD:v[0-9]+]], [[A]], [[B]] 9 ; VI-NEXT: buffer_store_short [[ADD]] 10 define amdgpu_kernel void @v_test_sub_i16(i16 addrspace(1)* %out, i16 addrspace(1)* %in0, i16 addrspace(1)* %in1) #1 { 11 %tid = call i32 @llvm.amdgcn.workitem.id.x() 12 %gep.out = getelementptr inbounds i16, i16 addrspace(1)* %out, i32 %tid 13 %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid 14 %gep.in1 = getelementptr inbounds i16, i16 addrspace(1)* %in1, i32 %tid 15 %a = load volatile i16, i16 addrspace(1)* %gep.in0 16 %b = load volatile i16, i16 addrspace(1)* %gep.in1 17 %add = sub i16 %a, %b 18 store i16 %add, i16 addrspace(1)* %out 19 ret void 20 } 21 22 ; FIXME: Need to handle non-uniform case for function below (load without gep). 23 ; GCN-LABEL: {{^}}v_test_sub_i16_constant: 24 ; VI: flat_load_ushort [[A:v[0-9]+]] 25 ; VI: v_add_u16_e32 [[ADD:v[0-9]+]], 0xffffff85, [[A]] 26 ; VI-NEXT: buffer_store_short [[ADD]] 27 define amdgpu_kernel void @v_test_sub_i16_constant(i16 addrspace(1)* %out, i16 addrspace(1)* %in0) #1 { 28 %tid = call i32 @llvm.amdgcn.workitem.id.x() 29 %gep.out = getelementptr inbounds i16, i16 addrspace(1)* %out, i32 %tid 30 %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid 31 %a = load volatile i16, i16 addrspace(1)* %gep.in0 32 %add = sub i16 %a, 123 33 store i16 %add, i16 addrspace(1)* %out 34 ret void 35 } 36 37 ; FIXME: Need to handle non-uniform case for function below (load without gep). 38 ; GCN-LABEL: {{^}}v_test_sub_i16_neg_constant: 39 ; VI: flat_load_ushort [[A:v[0-9]+]] 40 ; VI: v_add_u16_e32 [[ADD:v[0-9]+]], 0x34d, [[A]] 41 ; VI-NEXT: buffer_store_short [[ADD]] 42 define amdgpu_kernel void @v_test_sub_i16_neg_constant(i16 addrspace(1)* %out, i16 addrspace(1)* %in0) #1 { 43 %tid = call i32 @llvm.amdgcn.workitem.id.x() 44 %gep.out = getelementptr inbounds i16, i16 addrspace(1)* %out, i32 %tid 45 %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid 46 %a = load volatile i16, i16 addrspace(1)* %gep.in0 47 %add = sub i16 %a, -845 48 store i16 %add, i16 addrspace(1)* %out 49 ret void 50 } 51 52 ; FIXME: Need to handle non-uniform case for function below (load without gep). 53 ; GCN-LABEL: {{^}}v_test_sub_i16_inline_63: 54 ; VI: flat_load_ushort [[A:v[0-9]+]] 55 ; VI: v_subrev_u16_e32 [[ADD:v[0-9]+]], 63, [[A]] 56 ; VI-NEXT: buffer_store_short [[ADD]] 57 define amdgpu_kernel void @v_test_sub_i16_inline_63(i16 addrspace(1)* %out, i16 addrspace(1)* %in0) #1 { 58 %tid = call i32 @llvm.amdgcn.workitem.id.x() 59 %gep.out = getelementptr inbounds i16, i16 addrspace(1)* %out, i32 %tid 60 %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid 61 %a = load volatile i16, i16 addrspace(1)* %gep.in0 62 %add = sub i16 %a, 63 63 store i16 %add, i16 addrspace(1)* %out 64 ret void 65 } 66 67 ; FIXME: Need to handle non-uniform case for function below (load without gep). 68 ; GCN-LABEL: {{^}}v_test_sub_i16_zext_to_i32: 69 ; VI: flat_load_ushort [[A:v[0-9]+]] 70 ; VI: flat_load_ushort [[B:v[0-9]+]] 71 ; VI: v_sub_u16_e32 [[ADD:v[0-9]+]], [[A]], [[B]] 72 ; VI-NEXT: buffer_store_dword [[ADD]] 73 define amdgpu_kernel void @v_test_sub_i16_zext_to_i32(i32 addrspace(1)* %out, i16 addrspace(1)* %in0, i16 addrspace(1)* %in1) #1 { 74 %tid = call i32 @llvm.amdgcn.workitem.id.x() 75 %gep.out = getelementptr inbounds i32, i32 addrspace(1)* %out, i32 %tid 76 %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid 77 %gep.in1 = getelementptr inbounds i16, i16 addrspace(1)* %in1, i32 %tid 78 %a = load volatile i16, i16 addrspace(1)* %gep.in0 79 %b = load volatile i16, i16 addrspace(1)* %gep.in1 80 %add = sub i16 %a, %b 81 %ext = zext i16 %add to i32 82 store i32 %ext, i32 addrspace(1)* %out 83 ret void 84 } 85 86 ; FIXME: Need to handle non-uniform case for function below (load without gep). 87 ; GCN-LABEL: {{^}}v_test_sub_i16_zext_to_i64: 88 ; VI: flat_load_ushort [[A:v[0-9]+]] 89 ; VI: flat_load_ushort [[B:v[0-9]+]] 90 ; VI: v_mov_b32_e32 v[[VZERO:[0-9]+]], 0 91 ; VI-DAG: v_sub_u16_e32 v[[ADD:[0-9]+]], [[A]], [[B]] 92 ; VI: buffer_store_dwordx2 v{{\[}}[[ADD]]:[[VZERO]]{{\]}}, off, {{s\[[0-9]+:[0-9]+\]}}, 0{{$}} 93 define amdgpu_kernel void @v_test_sub_i16_zext_to_i64(i64 addrspace(1)* %out, i16 addrspace(1)* %in0, i16 addrspace(1)* %in1) #1 { 94 %tid = call i32 @llvm.amdgcn.workitem.id.x() 95 %gep.out = getelementptr inbounds i64, i64 addrspace(1)* %out, i32 %tid 96 %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid 97 %gep.in1 = getelementptr inbounds i16, i16 addrspace(1)* %in1, i32 %tid 98 %a = load volatile i16, i16 addrspace(1)* %gep.in0 99 %b = load volatile i16, i16 addrspace(1)* %gep.in1 100 %add = sub i16 %a, %b 101 %ext = zext i16 %add to i64 102 store i64 %ext, i64 addrspace(1)* %out 103 ret void 104 } 105 106 ; FIXME: Need to handle non-uniform case for function below (load without gep). 107 ; GCN-LABEL: {{^}}v_test_sub_i16_sext_to_i32: 108 ; VI: flat_load_ushort [[A:v[0-9]+]] 109 ; VI: flat_load_ushort [[B:v[0-9]+]] 110 ; VI: v_sub_u16_e32 [[ADD:v[0-9]+]], [[B]], [[A]] 111 ; VI-NEXT: v_bfe_i32 [[SEXT:v[0-9]+]], [[ADD]], 0, 16 112 ; VI-NEXT: buffer_store_dword [[SEXT]] 113 define amdgpu_kernel void @v_test_sub_i16_sext_to_i32(i32 addrspace(1)* %out, i16 addrspace(1)* %in0, i16 addrspace(1)* %in1) #1 { 114 %tid = call i32 @llvm.amdgcn.workitem.id.x() 115 %gep.out = getelementptr inbounds i32, i32 addrspace(1)* %out, i32 %tid 116 %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid 117 %gep.in1 = getelementptr inbounds i16, i16 addrspace(1)* %in1, i32 %tid 118 %a = load i16, i16 addrspace(1)* %gep.in0 119 %b = load i16, i16 addrspace(1)* %gep.in1 120 %add = sub i16 %a, %b 121 %ext = sext i16 %add to i32 122 store i32 %ext, i32 addrspace(1)* %out 123 ret void 124 } 125 126 ; FIXME: Need to handle non-uniform case for function below (load without gep). 127 ; GCN-LABEL: {{^}}v_test_sub_i16_sext_to_i64: 128 ; VI: flat_load_ushort [[A:v[0-9]+]] 129 ; VI: flat_load_ushort [[B:v[0-9]+]] 130 ; VI: v_sub_u16_e32 [[ADD:v[0-9]+]], [[B]], [[A]] 131 ; VI-NEXT: v_bfe_i32 v[[LO:[0-9]+]], [[ADD]], 0, 16 132 ; VI-NEXT: v_ashrrev_i32_e32 v[[HI:[0-9]+]], 31, v[[LO]] 133 ; VI-NEXT: buffer_store_dwordx2 v{{\[}}[[LO]]:[[HI]]{{\]}} 134 define amdgpu_kernel void @v_test_sub_i16_sext_to_i64(i64 addrspace(1)* %out, i16 addrspace(1)* %in0, i16 addrspace(1)* %in1) #1 { 135 %tid = call i32 @llvm.amdgcn.workitem.id.x() 136 %gep.out = getelementptr inbounds i64, i64 addrspace(1)* %out, i32 %tid 137 %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid 138 %gep.in1 = getelementptr inbounds i16, i16 addrspace(1)* %in1, i32 %tid 139 %a = load i16, i16 addrspace(1)* %gep.in0 140 %b = load i16, i16 addrspace(1)* %gep.in1 141 %add = sub i16 %a, %b 142 %ext = sext i16 %add to i64 143 store i64 %ext, i64 addrspace(1)* %out 144 ret void 145 } 146 147 @lds = addrspace(3) global [512 x i32] undef, align 4 148 149 ; GCN-LABEL: {{^}}v_test_sub_i16_constant_commute: 150 ; VI: v_subrev_u16_e32 v{{[0-9]+}}, 0x800, v{{[0-9]+}} 151 ; CI: v_subrev_i32_e32 v{{[0-9]+}}, vcc, 0x800, v{{[0-9]+}} 152 define amdgpu_kernel void @v_test_sub_i16_constant_commute(i16 addrspace(1)* %out, i16 addrspace(1)* %in0) #1 { 153 %size = call i32 @llvm.amdgcn.groupstaticsize() 154 %size.trunc = trunc i32 %size to i16 155 call void asm sideeffect "; $0", "v"([512 x i32] addrspace(3)* @lds) 156 %tid = call i32 @llvm.amdgcn.workitem.id.x() 157 %gep.out = getelementptr inbounds i16, i16 addrspace(1)* %out, i32 %tid 158 %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid 159 %a = load volatile i16, i16 addrspace(1)* %gep.in0 160 %add = sub i16 %a, %size.trunc 161 store i16 %add, i16 addrspace(1)* %out 162 ret void 163 } 164 165 declare i32 @llvm.amdgcn.workitem.id.x() #0 166 declare i32 @llvm.amdgcn.groupstaticsize() #0 167 168 attributes #0 = { nounwind readnone } 169 attributes #1 = { nounwind } 170