diff options
| author | Matt Arsenault <Matthew.Arsenault@amd.com> | 2017-02-22 23:04:58 +0000 |
|---|---|---|
| committer | Matt Arsenault <Matthew.Arsenault@amd.com> | 2017-02-22 23:04:58 +0000 |
| commit | f5262256a1380fec75c7fa7023df52a459f0d6f0 (patch) | |
| tree | b318bcc1e270982f69891b39fcb65a5758c70ad0 /llvm/test/CodeGen | |
| parent | 4805ce0b178a0e9f8f7ca618c78cc810fd37e42d (diff) | |
| download | bcm5719-llvm-f5262256a1380fec75c7fa7023df52a459f0d6f0.tar.gz bcm5719-llvm-f5262256a1380fec75c7fa7023df52a459f0d6f0.zip | |
AMDGPU: Add replacement bfe intrinsics
llvm-svn: 295899
Diffstat (limited to 'llvm/test/CodeGen')
| -rw-r--r-- | llvm/test/CodeGen/AMDGPU/llvm.amdgcn.sbfe.ll | 413 | ||||
| -rw-r--r-- | llvm/test/CodeGen/AMDGPU/llvm.amdgcn.ubfe.ll | 623 |
2 files changed, 1036 insertions, 0 deletions
diff --git a/llvm/test/CodeGen/AMDGPU/llvm.amdgcn.sbfe.ll b/llvm/test/CodeGen/AMDGPU/llvm.amdgcn.sbfe.ll new file mode 100644 index 00000000000..c51c1c957f8 --- /dev/null +++ b/llvm/test/CodeGen/AMDGPU/llvm.amdgcn.sbfe.ll @@ -0,0 +1,413 @@ +; RUN: llc -march=amdgcn -verify-machineinstrs < %s | FileCheck -check-prefix=GCN -check-prefix=SI %s +; RUN: llc -march=amdgcn -mcpu=tonga -mattr=-flat-for-global -verify-machineinstrs < %s | FileCheck -check-prefix=GCN -check-prefix=VI %s + +; GCN-LABEL: {{^}}bfe_i32_arg_arg_arg: +; GCN: v_bfe_i32 +define void @bfe_i32_arg_arg_arg(i32 addrspace(1)* %out, i32 %src0, i32 %src1, i32 %src2) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 %src0, i32 %src1, i32 %src1) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_arg_arg_imm: +; GCN: v_bfe_i32 +define void @bfe_i32_arg_arg_imm(i32 addrspace(1)* %out, i32 %src0, i32 %src1) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 %src0, i32 %src1, i32 123) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_arg_imm_arg: +; GCN: v_bfe_i32 +define void @bfe_i32_arg_imm_arg(i32 addrspace(1)* %out, i32 %src0, i32 %src2) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 %src0, i32 123, i32 %src2) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_imm_arg_arg: +; GCN: v_bfe_i32 +define void @bfe_i32_imm_arg_arg(i32 addrspace(1)* %out, i32 %src1, i32 %src2) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 123, i32 %src1, i32 %src2) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}v_bfe_print_arg: +; GCN: v_bfe_i32 v{{[0-9]+}}, v{{[0-9]+}}, 2, 8 +define void @v_bfe_print_arg(i32 addrspace(1)* %out, i32 addrspace(1)* %src0) #0 { + %load = load i32, i32 addrspace(1)* %src0, align 4 + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 %load, i32 2, i32 8) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_arg_0_width_reg_offset: +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_i32_arg_0_width_reg_offset(i32 addrspace(1)* %out, i32 %src0, i32 %src1) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.sbfe.i32(i32 %src0, i32 %src1, i32 0) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_arg_0_width_imm_offset: +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_i32_arg_0_width_imm_offset(i32 addrspace(1)* %out, i32 %src0, i32 %src1) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.sbfe.i32(i32 %src0, i32 8, i32 0) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_test_6: +; GCN: v_lshlrev_b32_e32 v{{[0-9]+}}, 31, v{{[0-9]+}} +; GCN: v_ashrrev_i32_e32 v{{[0-9]+}}, 1, v{{[0-9]+}} +; GCN: s_endpgm +define void @bfe_i32_test_6(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = shl i32 %x, 31 + %bfe = call i32 @llvm.amdgcn.sbfe.i32(i32 %shl, i32 1, i32 31) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_test_7: +; GCN-NOT: shl +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_test_7(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = shl i32 %x, 31 + %bfe = call i32 @llvm.amdgcn.sbfe.i32(i32 %shl, i32 0, i32 31) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_test_8: +; GCN: buffer_load_dword +; GCN: v_bfe_i32 v{{[0-9]+}}, v{{[0-9]+}}, 0, 1 +; GCN: s_endpgm +define void @bfe_i32_test_8(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = shl i32 %x, 31 + %bfe = call i32 @llvm.amdgcn.sbfe.i32(i32 %shl, i32 31, i32 1) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_test_9: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_ashrrev_i32_e32 v{{[0-9]+}}, 31, v{{[0-9]+}} +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_i32_test_9(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %bfe = call i32 @llvm.amdgcn.sbfe.i32(i32 %x, i32 31, i32 1) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_test_10: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_ashrrev_i32_e32 v{{[0-9]+}}, 1, v{{[0-9]+}} +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_i32_test_10(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %bfe = call i32 @llvm.amdgcn.sbfe.i32(i32 %x, i32 1, i32 31) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_test_11: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_ashrrev_i32_e32 v{{[0-9]+}}, 8, v{{[0-9]+}} +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_i32_test_11(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %bfe = call i32 @llvm.amdgcn.sbfe.i32(i32 %x, i32 8, i32 24) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_test_12: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_ashrrev_i32_e32 v{{[0-9]+}}, 24, v{{[0-9]+}} +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_i32_test_12(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %bfe = call i32 @llvm.amdgcn.sbfe.i32(i32 %x, i32 24, i32 8) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_test_13: +; GCN: v_ashrrev_i32_e32 {{v[0-9]+}}, 31, {{v[0-9]+}} +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_i32_test_13(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = ashr i32 %x, 31 + %bfe = call i32 @llvm.amdgcn.sbfe.i32(i32 %shl, i32 31, i32 1) + store i32 %bfe, i32 addrspace(1)* %out, align 4 ret void +} + +; GCN-LABEL: {{^}}bfe_i32_test_14: +; GCN-NOT: lshr +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_i32_test_14(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = lshr i32 %x, 31 + %bfe = call i32 @llvm.amdgcn.sbfe.i32(i32 %shl, i32 31, i32 1) + store i32 %bfe, i32 addrspace(1)* %out, align 4 ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_0: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_0(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 0, i32 0, i32 0) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_1: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_1(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 12334, i32 0, i32 0) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_2: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_2(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 0, i32 0, i32 1) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_3: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], -1 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_3(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 1, i32 0, i32 1) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_4: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], -1 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_4(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 4294967295, i32 0, i32 1) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_5: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], -1 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_5(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 128, i32 7, i32 1) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_6: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0xffffff80 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_6(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 128, i32 0, i32 8) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_7: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0x7f +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_7(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 127, i32 0, i32 8) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_8: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 1 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_8(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 127, i32 6, i32 8) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_9: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 1 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_9(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 65536, i32 16, i32 8) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_10: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_10(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 65535, i32 16, i32 16) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_11: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], -6 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_11(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 160, i32 4, i32 4) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_12: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_12(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 160, i32 31, i32 1) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_13: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 1 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_13(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 131070, i32 16, i32 16) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_14: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 40 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_14(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 160, i32 2, i32 30) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_15: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 10 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_15(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 160, i32 4, i32 28) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_16: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], -1 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_16(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 4294967295, i32 1, i32 7) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_17: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0x7f +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_17(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 255, i32 1, i32 31) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_i32_constant_fold_test_18: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_i32_constant_fold_test_18(i32 addrspace(1)* %out) #0 { + %bfe_i32 = call i32 @llvm.amdgcn.sbfe.i32(i32 255, i32 31, i32 1) + store i32 %bfe_i32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_sext_in_reg_i24: +; GCN: buffer_load_dword [[LOAD:v[0-9]+]], +; GCN-NOT: v_lshl +; GCN-NOT: v_ashr +; GCN: v_bfe_i32 [[BFE:v[0-9]+]], [[LOAD]], 0, 24 +; GCN: buffer_store_dword [[BFE]], +define void @bfe_sext_in_reg_i24(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %bfe = call i32 @llvm.amdgcn.sbfe.i32(i32 %x, i32 0, i32 24) + %shl = shl i32 %bfe, 8 + %ashr = ashr i32 %shl, 8 + store i32 %ashr, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: @simplify_demanded_bfe_sdiv +; GCN: buffer_load_dword [[LOAD:v[0-9]+]] +; GCN: v_bfe_i32 [[BFE:v[0-9]+]], [[LOAD]], 1, 16 +; GCN: v_lshrrev_b32_e32 [[TMP0:v[0-9]+]], 31, [[BFE]] +; GCN: v_add_i32_e32 [[TMP1:v[0-9]+]], vcc, [[TMP0]], [[BFE]] +; GCN: v_ashrrev_i32_e32 [[TMP2:v[0-9]+]], 1, [[TMP1]] +; GCN: buffer_store_dword [[TMP2]] +define void @simplify_demanded_bfe_sdiv(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %src = load i32, i32 addrspace(1)* %in, align 4 + %bfe = call i32 @llvm.amdgcn.sbfe.i32(i32 %src, i32 1, i32 16) + %div = sdiv i32 %bfe, 2 + store i32 %div, i32 addrspace(1)* %out, align 4 + ret void +} + +declare i32 @llvm.amdgcn.sbfe.i32(i32, i32, i32) #1 + +attributes #0 = { nounwind } +attributes #1 = { nounwind readnone } diff --git a/llvm/test/CodeGen/AMDGPU/llvm.amdgcn.ubfe.ll b/llvm/test/CodeGen/AMDGPU/llvm.amdgcn.ubfe.ll new file mode 100644 index 00000000000..8a2990a8cbd --- /dev/null +++ b/llvm/test/CodeGen/AMDGPU/llvm.amdgcn.ubfe.ll @@ -0,0 +1,623 @@ +; RUN: llc -march=amdgcn -verify-machineinstrs < %s | FileCheck -check-prefix=GCN -check-prefix=SI %s +; RUN: llc -march=amdgcn -mcpu=tonga -mattr=-flat-for-global -verify-machineinstrs < %s | FileCheck -check-prefix=GCN -check-prefix=VI %s + +; GCN-LABEL: {{^}}bfe_u32_arg_arg_arg: +; GCN: v_bfe_u32 +define void @bfe_u32_arg_arg_arg(i32 addrspace(1)* %out, i32 %src0, i32 %src1, i32 %src2) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 %src0, i32 %src1, i32 %src1) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_arg_arg_imm: +; GCN: v_bfe_u32 +define void @bfe_u32_arg_arg_imm(i32 addrspace(1)* %out, i32 %src0, i32 %src1) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 %src0, i32 %src1, i32 123) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_arg_imm_arg: +; GCN: v_bfe_u32 +define void @bfe_u32_arg_imm_arg(i32 addrspace(1)* %out, i32 %src0, i32 %src2) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 %src0, i32 123, i32 %src2) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_imm_arg_arg: +; GCN: v_bfe_u32 +define void @bfe_u32_imm_arg_arg(i32 addrspace(1)* %out, i32 %src1, i32 %src2) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 123, i32 %src1, i32 %src2) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_arg_0_width_reg_offset: +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_u32_arg_0_width_reg_offset(i32 addrspace(1)* %out, i32 %src0, i32 %src1) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 %src0, i32 %src1, i32 0) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_arg_0_width_imm_offset: +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_u32_arg_0_width_imm_offset(i32 addrspace(1)* %out, i32 %src0, i32 %src1) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 %src0, i32 8, i32 0) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_zextload_i8: +; GCN: buffer_load_ubyte +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_u32_zextload_i8(i32 addrspace(1)* %out, i8 addrspace(1)* %in) #0 { + %load = load i8, i8 addrspace(1)* %in + %ext = zext i8 %load to i32 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %ext, i32 0, i32 8) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_zext_in_reg_i8: +; GCN: buffer_load_dword +; GCN: v_add_i32 +; GCN-NEXT: v_and_b32_e32 +; FIXME: Should be using s_add_i32 +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_u32_zext_in_reg_i8(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %load = load i32, i32 addrspace(1)* %in, align 4 + %add = add i32 %load, 1 + %ext = and i32 %add, 255 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %ext, i32 0, i32 8) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_zext_in_reg_i16: +; GCN: buffer_load_dword +; GCN: v_add_i32 +; GCN-NEXT: v_and_b32_e32 +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_u32_zext_in_reg_i16(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %load = load i32, i32 addrspace(1)* %in, align 4 + %add = add i32 %load, 1 + %ext = and i32 %add, 65535 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %ext, i32 0, i32 16) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_zext_in_reg_i8_offset_1: +; GCN: buffer_load_dword +; GCN: v_add_i32 +; GCN: bfe +; GCN: s_endpgm +define void @bfe_u32_zext_in_reg_i8_offset_1(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %load = load i32, i32 addrspace(1)* %in, align 4 + %add = add i32 %load, 1 + %ext = and i32 %add, 255 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %ext, i32 1, i32 8) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_zext_in_reg_i8_offset_3: +; GCN: buffer_load_dword +; GCN: v_add_i32 +; GCN-NEXT: v_and_b32_e32 {{v[0-9]+}}, 0xf8 +; GCN-NEXT: bfe +; GCN: s_endpgm +define void @bfe_u32_zext_in_reg_i8_offset_3(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %load = load i32, i32 addrspace(1)* %in, align 4 + %add = add i32 %load, 1 + %ext = and i32 %add, 255 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %ext, i32 3, i32 8) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_zext_in_reg_i8_offset_7: +; GCN: buffer_load_dword +; GCN: v_add_i32 +; GCN-NEXT: v_and_b32_e32 {{v[0-9]+}}, 0x80 +; GCN-NEXT: bfe +; GCN: s_endpgm +define void @bfe_u32_zext_in_reg_i8_offset_7(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %load = load i32, i32 addrspace(1)* %in, align 4 + %add = add i32 %load, 1 + %ext = and i32 %add, 255 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %ext, i32 7, i32 8) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_zext_in_reg_i16_offset_8: +; GCN: buffer_load_dword +; GCN: v_add_i32 +; GCN-NEXT: bfe +; GCN: s_endpgm +define void @bfe_u32_zext_in_reg_i16_offset_8(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %load = load i32, i32 addrspace(1)* %in, align 4 + %add = add i32 %load, 1 + %ext = and i32 %add, 65535 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %ext, i32 8, i32 8) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_test_1: +; GCN: buffer_load_dword +; GCN: v_and_b32_e32 {{v[0-9]+}}, 1, {{v[0-9]+}} +; GCN: s_endpgm +define void @bfe_u32_test_1(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %x, i32 0, i32 1) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +define void @bfe_u32_test_2(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = shl i32 %x, 31 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %shl, i32 0, i32 8) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +define void @bfe_u32_test_3(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = shl i32 %x, 31 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %shl, i32 0, i32 1) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_test_4: +; GCN-NOT: lshl +; GCN-NOT: shr +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +define void @bfe_u32_test_4(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = shl i32 %x, 31 + %shr = lshr i32 %shl, 31 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %shr, i32 31, i32 1) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_test_5: +; GCN: buffer_load_dword +; GCN-NOT: lshl +; GCN-NOT: shr +; GCN: v_bfe_i32 {{v[0-9]+}}, {{v[0-9]+}}, 0, 1 +; GCN: s_endpgm +define void @bfe_u32_test_5(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = shl i32 %x, 31 + %shr = ashr i32 %shl, 31 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %shr, i32 0, i32 1) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_test_6: +; GCN: v_lshlrev_b32_e32 v{{[0-9]+}}, 31, v{{[0-9]+}} +; GCN: v_lshrrev_b32_e32 v{{[0-9]+}}, 1, v{{[0-9]+}} +; GCN: s_endpgm +define void @bfe_u32_test_6(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = shl i32 %x, 31 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %shl, i32 1, i32 31) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_test_7: +; GCN: v_lshlrev_b32_e32 v{{[0-9]+}}, 31, v{{[0-9]+}} +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_u32_test_7(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = shl i32 %x, 31 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %shl, i32 0, i32 31) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_test_8: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_and_b32_e32 {{v[0-9]+}}, 1, {{v[0-9]+}} +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_u32_test_8(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = shl i32 %x, 31 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %shl, i32 31, i32 1) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_test_9: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_lshrrev_b32_e32 v{{[0-9]+}}, 31, v{{[0-9]+}} +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_u32_test_9(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %x, i32 31, i32 1) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_test_10: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_lshrrev_b32_e32 v{{[0-9]+}}, 1, v{{[0-9]+}} +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_u32_test_10(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %x, i32 1, i32 31) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_test_11: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_lshrrev_b32_e32 v{{[0-9]+}}, 8, v{{[0-9]+}} +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_u32_test_11(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %x, i32 8, i32 24) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_test_12: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_lshrrev_b32_e32 v{{[0-9]+}}, 24, v{{[0-9]+}} +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_u32_test_12(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %x, i32 24, i32 8) + store i32 %bfe, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_test_13: +; V_ASHRREV_U32_e32 {{v[0-9]+}}, 31, {{v[0-9]+}} +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_u32_test_13(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = ashr i32 %x, 31 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %shl, i32 31, i32 1) + store i32 %bfe, i32 addrspace(1)* %out, align 4 ret void +} + +; GCN-LABEL: {{^}}bfe_u32_test_14: +; GCN-NOT: lshr +; GCN-NOT: {{[^@]}}bfe +; GCN: s_endpgm +define void @bfe_u32_test_14(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 { + %x = load i32, i32 addrspace(1)* %in, align 4 + %shl = lshr i32 %x, 31 + %bfe = call i32 @llvm.amdgcn.ubfe.i32(i32 %shl, i32 31, i32 1) + store i32 %bfe, i32 addrspace(1)* %out, align 4 ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_0: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_0(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 0, i32 0, i32 0) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_1: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_1(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 12334, i32 0, i32 0) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_2: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_2(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 0, i32 0, i32 1) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_3: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 1 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_3(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 1, i32 0, i32 1) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_4: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], -1 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_4(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 4294967295, i32 0, i32 1) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_5: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 1 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_5(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 128, i32 7, i32 1) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_6: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0x80 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_6(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 128, i32 0, i32 8) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_7: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0x7f +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_7(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 127, i32 0, i32 8) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_8: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 1 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_8(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 127, i32 6, i32 8) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_9: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 1 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_9(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 65536, i32 16, i32 8) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_10: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_10(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 65535, i32 16, i32 16) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_11: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 10 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_11(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 160, i32 4, i32 4) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_12: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_12(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 160, i32 31, i32 1) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_13: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 1 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_13(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 131070, i32 16, i32 16) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_14: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 40 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_14(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 160, i32 2, i32 30) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_15: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 10 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_15(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 160, i32 4, i32 28) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_16: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0x7f +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_16(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 4294967295, i32 1, i32 7) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_17: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0x7f +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_17(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 255, i32 1, i32 31) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; GCN-LABEL: {{^}}bfe_u32_constant_fold_test_18: +; GCN-NOT: {{[^@]}}bfe +; GCN: v_mov_b32_e32 [[VREG:v[0-9]+]], 0 +; GCN: buffer_store_dword [[VREG]], +; GCN: s_endpgm +; EG-NOT: BFE +define void @bfe_u32_constant_fold_test_18(i32 addrspace(1)* %out) #0 { + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 255, i32 31, i32 1) + store i32 %bfe_u32, i32 addrspace(1)* %out, align 4 + ret void +} + +; Make sure that SimplifyDemandedBits doesn't cause the and to be +; reduced to the bits demanded by the bfe. + +; XXX: The operand to v_bfe_u32 could also just directly be the load register. +; GCN-LABEL: {{^}}simplify_bfe_u32_multi_use_arg: +; GCN: buffer_load_dword [[ARG:v[0-9]+]] +; GCN: v_and_b32_e32 [[AND:v[0-9]+]], 63, [[ARG]] +; GCN: v_bfe_u32 [[BFE:v[0-9]+]], [[AND]], 2, 2 +; GCN-DAG: buffer_store_dword [[AND]] +; GCN-DAG: buffer_store_dword [[BFE]] +; GCN: s_endpgm +define void @simplify_bfe_u32_multi_use_arg(i32 addrspace(1)* %out0, + i32 addrspace(1)* %out1, + i32 addrspace(1)* %in) #0 { + %src = load i32, i32 addrspace(1)* %in, align 4 + %and = and i32 %src, 63 + %bfe_u32 = call i32 @llvm.amdgcn.ubfe.i32(i32 %and, i32 2, i32 2) + store i32 %bfe_u32, i32 addrspace(1)* %out0, align 4 + store i32 %and, i32 addrspace(1)* %out1, align 4 + ret void +} + +; GCN-LABEL: {{^}}lshr_and: +; GCN: s_bfe_u32 {{s[0-9]+}}, {{s[0-9]+}}, 0x30006 +; GCN: buffer_store_dword +define void @lshr_and(i32 addrspace(1)* %out, i32 %a) #0 { + %b = lshr i32 %a, 6 + %c = and i32 %b, 7 + store i32 %c, i32 addrspace(1)* %out, align 8 + ret void +} + +; GCN-LABEL: {{^}}v_lshr_and: +; GCN: v_bfe_u32 {{v[0-9]+}}, {{s[0-9]+}}, {{v[0-9]+}}, 3 +; GCN: buffer_store_dword +define void @v_lshr_and(i32 addrspace(1)* %out, i32 %a, i32 %b) #0 { + %c = lshr i32 %a, %b + %d = and i32 %c, 7 + store i32 %d, i32 addrspace(1)* %out, align 8 + ret void +} + +; GCN-LABEL: {{^}}and_lshr: +; GCN: s_bfe_u32 {{s[0-9]+}}, {{s[0-9]+}}, 0x30006 +; GCN: buffer_store_dword +define void @and_lshr(i32 addrspace(1)* %out, i32 %a) #0 { + %b = and i32 %a, 448 + %c = lshr i32 %b, 6 + store i32 %c, i32 addrspace(1)* %out, align 8 + ret void +} + +; GCN-LABEL: {{^}}and_lshr2: +; GCN: s_bfe_u32 {{s[0-9]+}}, {{s[0-9]+}}, 0x30006 +; GCN: buffer_store_dword +define void @and_lshr2(i32 addrspace(1)* %out, i32 %a) #0 { + %b = and i32 %a, 511 + %c = lshr i32 %b, 6 + store i32 %c, i32 addrspace(1)* %out, align 8 + ret void +} + +; GCN-LABEL: {{^}}shl_lshr: +; GCN: s_bfe_u32 {{s[0-9]+}}, {{s[0-9]+}}, 0x150002 +; GCN: buffer_store_dword +define void @shl_lshr(i32 addrspace(1)* %out, i32 %a) #0 { + %b = shl i32 %a, 9 + %c = lshr i32 %b, 11 + store i32 %c, i32 addrspace(1)* %out, align 8 + ret void +} + +declare i32 @llvm.amdgcn.ubfe.i32(i32, i32, i32) #1 + +attributes #0 = { nounwind } +attributes #1 = { nounwind readnone } |

