diff --git a/llvm/lib/Target/AMDGPU/VOP2Instructions.td b/llvm/lib/Target/AMDGPU/VOP2Instructions.td index b87f3be21399014c62eade186d197410b9234b39..dec133c09cacfdad7b2af53e121969ddccbed600 100644 --- a/llvm/lib/Target/AMDGPU/VOP2Instructions.td +++ b/llvm/lib/Target/AMDGPU/VOP2Instructions.td @@ -424,9 +424,20 @@ defm : Arithmetic_i16_Pats; defm : Arithmetic_i16_Pats; defm : Arithmetic_i16_Pats; -defm : Arithmetic_i16_Pats; -defm : Arithmetic_i16_Pats; -defm : Arithmetic_i16_Pats; +def : Pat < + (and i16:$src0, i16:$src1), + (V_AND_B32_e32 $src0, $src1) +>; + +def : Pat < + (or i16:$src0, i16:$src1), + (V_OR_B32_e32 $src0, $src1) +>; + +def : Pat < + (xor i16:$src0, i16:$src1), + (V_XOR_B32_e32 $src0, $src1) +>; defm : Bits_OpsRev_i16_Pats; defm : Bits_OpsRev_i16_Pats; diff --git a/llvm/test/CodeGen/AMDGPU/extend-bit-ops-i16.ll b/llvm/test/CodeGen/AMDGPU/extend-bit-ops-i16.ll new file mode 100644 index 0000000000000000000000000000000000000000..cf384da2c5beb893c32805aa4b10d9af6430ed44 --- /dev/null +++ b/llvm/test/CodeGen/AMDGPU/extend-bit-ops-i16.ll @@ -0,0 +1,50 @@ +; RUN: llc < %s -march=amdgcn -mcpu=tonga -verify-machineinstrs | FileCheck %s --check-prefix=GCN + +; GCN-LABEL: and_zext: +; GCN: v_and_b32_e32 [[VAL16:v[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}} +; GCN: v_and_b32_e32 v{{[0-9]+}}, 0xffff, [[VAL16]] +define void @and_zext(i32 addrspace(1)* %out, i16 addrspace(1)* %in) { + %id = call i32 @llvm.amdgcn.workitem.id.x() #1 + %ptr = getelementptr i16, i16 addrspace(1)* %in, i32 %id + %a = load i16, i16 addrspace(1)* %in + %b = load i16, i16 addrspace(1)* %ptr + %c = add i16 %a, %b + %val16 = and i16 %c, %a + %val32 = zext i16 %val16 to i32 + store i32 %val32, i32 addrspace(1)* %out + ret void +} + +; GCN-LABEL: or_zext: +; GCN: v_or_b32_e32 [[VAL16:v[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}} +; GCN: v_and_b32_e32 v{{[0-9]+}}, 0xffff, [[VAL16]] +define void @or_zext(i32 addrspace(1)* %out, i16 addrspace(1)* %in) { + %id = call i32 @llvm.amdgcn.workitem.id.x() #1 + %ptr = getelementptr i16, i16 addrspace(1)* %in, i32 %id + %a = load i16, i16 addrspace(1)* %in + %b = load i16, i16 addrspace(1)* %ptr + %c = add i16 %a, %b + %val16 = or i16 %c, %a + %val32 = zext i16 %val16 to i32 + store i32 %val32, i32 addrspace(1)* %out + ret void +} + +; GCN-LABEL: xor_zext: +; GCN: v_xor_b32_e32 [[VAL16:v[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}} +; GCN: v_and_b32_e32 v{{[0-9]+}}, 0xffff, [[VAL16]] +define void @xor_zext(i32 addrspace(1)* %out, i16 addrspace(1)* %in) { + %id = call i32 @llvm.amdgcn.workitem.id.x() #1 + %ptr = getelementptr i16, i16 addrspace(1)* %in, i32 %id + %a = load i16, i16 addrspace(1)* %in + %b = load i16, i16 addrspace(1)* %ptr + %c = add i16 %a, %b + %val16 = xor i16 %c, %a + %val32 = zext i16 %val16 to i32 + store i32 %val32, i32 addrspace(1)* %out + ret void +} + +declare i32 @llvm.amdgcn.workitem.id.x() #1 + +attributes #1 = { nounwind readnone }