1; RUN: llc -mtriple=amdgcn--amdhsa -mcpu=tonga -mattr=-flat-for-global -verify-machineinstrs < %s | FileCheck -check-prefix=VI -check-prefix=GCN %s 2; RUN: llc -mtriple=amdgcn--amdhsa -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]] 10define 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]+]], 0xff85, [[A]] 26; VI-NEXT: buffer_store_short [[ADD]] 27define 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]] 42define 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]] 57define 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]] 73define 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{{$}} 93define 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]+]], [[A]], [[B]] 111; VI: v_bfe_i32 [[SEXT:v[0-9]+]], [[ADD]], 0, 16 112; VI-NEXT: buffer_store_dword [[SEXT]] 113define 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]+]], [[A]], [[B]] 131; VI-NEXT: v_bfe_i32 v[[LO:[0-9]+]], [[ADD]], 0, 16 132; VI: v_ashrrev_i32_e32 v[[HI:[0-9]+]], 31, v[[LO]] 133; VI-NEXT: buffer_store_dwordx2 v{{\[}}[[LO]]:[[HI]]{{\]}} 134define 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]+}} 152define 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 165declare i32 @llvm.amdgcn.workitem.id.x() #0 166declare i32 @llvm.amdgcn.groupstaticsize() #0 167 168attributes #0 = { nounwind readnone } 169attributes #1 = { nounwind } 170