derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
184 lines
7.2 KiB
LLVM
184 lines
7.2 KiB
LLVM
; RUN: llc -mtriple=amdgcn < %s | FileCheck -check-prefix=GCN %s
|
|
|
|
; Make sure to test with f32 and i32 compares. If we have to use float
|
|
; compares, we always have multiple condition registers. If we can do
|
|
; scalar compares, we don't want to use multiple condition registers.
|
|
|
|
; GCN-LABEL: {{^}}opt_select_i32_and_cmp_i32:
|
|
; GCN-DAG: s_cmp_lg_u32
|
|
; GCN: s_cselect_b64 [[CMP1:s\[[0-9]+:[0-9]+\]]], -1, 0
|
|
; GCN-DAG: s_cmp_lg_u32
|
|
; GCN: s_cselect_b64 [[CMP2:s\[[0-9]+:[0-9]+\]]], -1, 0
|
|
; GCN: s_and_b64 [[AND1:s\[[0-9]+:[0-9]+\]]], [[CMP1]], [[CMP2]]
|
|
; GCN: s_and_b64 [[AND2:s\[[0-9]+:[0-9]+\]]], [[AND1]], exec
|
|
; GCN: s_cselect_b32 [[RESULT:s[0-9]+]]
|
|
; GCN: v_mov_b32_e32 [[VRESULT:v[0-9]+]], [[RESULT]]
|
|
; GCN: buffer_store_dword [[VRESULT]]
|
|
define amdgpu_kernel void @opt_select_i32_and_cmp_i32(ptr addrspace(1) %out, i32 %a, i32 %b, i32 %c, i32 %x, i32 %y) #0 {
|
|
%icmp0 = icmp ne i32 %a, %b
|
|
%icmp1 = icmp ne i32 %a, %c
|
|
%and = and i1 %icmp0, %icmp1
|
|
%select = select i1 %and, i32 %x, i32 %y
|
|
store i32 %select, ptr addrspace(1) %out
|
|
ret void
|
|
}
|
|
|
|
; GCN-LABEL: {{^}}opt_select_i32_and_cmp_f32:
|
|
; GCN-DAG: v_cmp_lg_f32_e32 vcc
|
|
; GCN-DAG: v_cmp_lg_f32_e64 [[CMP1:s\[[0-9]+:[0-9]+\]]]
|
|
; GCN: s_and_b64 [[CMP1]], vcc, [[CMP1]]
|
|
; GCN: s_and_b64 [[AND:s\[[0-9]+:[0-9]+\]]], [[CMP1]], exec
|
|
; GCN: s_cselect_b32 [[RESULT:s[0-9]+]]
|
|
; GCN: v_mov_b32_e32 [[VRESULT:v[0-9]+]], [[RESULT]]
|
|
; GCN: buffer_store_dword [[VRESULT]]
|
|
define amdgpu_kernel void @opt_select_i32_and_cmp_f32(ptr addrspace(1) %out, float %a, float %b, float %c, i32 %x, i32 %y) #0 {
|
|
%fcmp0 = fcmp one float %a, %b
|
|
%fcmp1 = fcmp one float %a, %c
|
|
%and = and i1 %fcmp0, %fcmp1
|
|
%select = select i1 %and, i32 %x, i32 %y
|
|
store i32 %select, ptr addrspace(1) %out
|
|
ret void
|
|
}
|
|
|
|
; GCN-LABEL: {{^}}opt_select_i64_and_cmp_i32:
|
|
; GCN-DAG: s_cmp_lg_u32
|
|
; GCN: s_cselect_b64 [[CMP1:s\[[0-9]+:[0-9]+\]]], -1, 0
|
|
; GCN-DAG: s_cmp_lg_u32
|
|
; GCN: s_cselect_b64 [[CMP2:s\[[0-9]+:[0-9]+\]]], -1, 0
|
|
; GCN: s_and_b64 [[AND1:s\[[0-9]+:[0-9]+\]]], [[CMP1]], [[CMP2]]
|
|
; GCN: s_and_b64 [[AND2:s\[[0-9]+:[0-9]+\]]], [[AND1]], exec
|
|
; GCN-DAG: s_cselect_b32 [[RESULT0:s[0-9]+]]
|
|
; GCN-DAG: s_cselect_b32 [[RESULT1:s[0-9]+]]
|
|
; GCN-DAG: v_mov_b32_e32 v[[VRESULT1:[0-9]+]], [[RESULT0]]
|
|
; GCN-DAG: v_mov_b32_e32 v[[VRESULT0:[0-9]+]], [[RESULT1]]
|
|
; GCN: buffer_store_dwordx2 v[[[VRESULT0]]:[[VRESULT1]]]
|
|
define amdgpu_kernel void @opt_select_i64_and_cmp_i32(ptr addrspace(1) %out, i32 %a, i32 %b, i32 %c, i64 %x, i64 %y) #0 {
|
|
%icmp0 = icmp ne i32 %a, %b
|
|
%icmp1 = icmp ne i32 %a, %c
|
|
%and = and i1 %icmp0, %icmp1
|
|
%select = select i1 %and, i64 %x, i64 %y
|
|
store i64 %select, ptr addrspace(1) %out
|
|
ret void
|
|
}
|
|
|
|
; GCN-LABEL: {{^}}opt_select_i64_and_cmp_f32:
|
|
; GCN-DAG: v_cmp_lg_f32_e32 vcc,
|
|
; GCN-DAG: v_cmp_lg_f32_e64 [[CMP1:s\[[0-9]+:[0-9]+\]]]
|
|
; GCN: s_and_b64 [[AND1:s\[[0-9]+:[0-9]+\]]], vcc, [[CMP1]]
|
|
; GCN: s_and_b64 [[AND2:s\[[0-9]+:[0-9]+\]]], [[AND1]], exec
|
|
; GCN-DAG: s_cselect_b32 [[RESULT0:s[0-9]+]]
|
|
; GCN-DAG: s_cselect_b32 [[RESULT1:s[0-9]+]]
|
|
; GCN-DAG: v_mov_b32_e32 v[[VRESULT1:[0-9]+]], [[RESULT0]]
|
|
; GCN-DAG: v_mov_b32_e32 v[[VRESULT0:[0-9]+]], [[RESULT1]]
|
|
; GCN: buffer_store_dwordx2 v[[[VRESULT0]]:[[VRESULT1]]]
|
|
define amdgpu_kernel void @opt_select_i64_and_cmp_f32(ptr addrspace(1) %out, float %a, float %b, float %c, i64 %x, i64 %y) #0 {
|
|
%fcmp0 = fcmp one float %a, %b
|
|
%fcmp1 = fcmp one float %a, %c
|
|
%and = and i1 %fcmp0, %fcmp1
|
|
%select = select i1 %and, i64 %x, i64 %y
|
|
store i64 %select, ptr addrspace(1) %out
|
|
ret void
|
|
}
|
|
|
|
; GCN-LABEL: {{^}}opt_select_i32_or_cmp_i32:
|
|
; GCN-DAG: s_cmp_lg_u32
|
|
; GCN: s_cselect_b64 [[CMP1:s\[[0-9]+:[0-9]+\]]], -1, 0
|
|
; GCN-DAG: s_cmp_lg_u32
|
|
; GCN: s_cselect_b64 [[CMP2:s\[[0-9]+:[0-9]+\]]], -1, 0
|
|
; GCN: s_or_b64 [[OR:s\[[0-9]+:[0-9]+\]]], [[CMP1]], [[CMP2]]
|
|
; GCN: s_and_b64 [[AND:s\[[0-9]+:[0-9]+\]]], [[OR]], exec
|
|
; GCN-DAG: s_cselect_b32 [[RESULT:s[0-9]+]]
|
|
; GCN-DAG: v_mov_b32_e32 [[VRESULT:v[0-9]+]], [[RESULT]]
|
|
; GCN: buffer_store_dword [[VRESULT]]
|
|
; GCN: s_endpgm
|
|
define amdgpu_kernel void @opt_select_i32_or_cmp_i32(ptr addrspace(1) %out, i32 %a, i32 %b, i32 %c, i32 %x, i32 %y) #0 {
|
|
%icmp0 = icmp ne i32 %a, %b
|
|
%icmp1 = icmp ne i32 %a, %c
|
|
%or = or i1 %icmp0, %icmp1
|
|
%select = select i1 %or, i32 %x, i32 %y
|
|
store i32 %select, ptr addrspace(1) %out
|
|
ret void
|
|
}
|
|
|
|
; GCN-LABEL: {{^}}opt_select_i32_or_cmp_f32:
|
|
; GCN-DAG: v_cmp_lg_f32_e32 vcc
|
|
; GCN-DAG: v_cmp_lg_f32_e64 [[CMP1:s\[[0-9]+:[0-9]+\]]]
|
|
; GCN: s_or_b64 [[OR:s\[[0-9]+:[0-9]+\]]], vcc, [[CMP1]]
|
|
; GCN: s_and_b64 [[AND:s\[[0-9]+:[0-9]+\]]], [[OR]], exec
|
|
; GCN-DAG: s_cselect_b32 [[RESULT:s[0-9]+]]
|
|
; GCN-DAG: v_mov_b32_e32 [[VRESULT:v[0-9]+]], [[RESULT]]
|
|
; GCN: buffer_store_dword [[VRESULT]]
|
|
define amdgpu_kernel void @opt_select_i32_or_cmp_f32(ptr addrspace(1) %out, float %a, float %b, float %c, i32 %x, i32 %y) #0 {
|
|
%fcmp0 = fcmp one float %a, %b
|
|
%fcmp1 = fcmp one float %a, %c
|
|
%or = or i1 %fcmp0, %fcmp1
|
|
%select = select i1 %or, i32 %x, i32 %y
|
|
store i32 %select, ptr addrspace(1) %out
|
|
ret void
|
|
}
|
|
|
|
; GCN-LABEL: {{^}}opt_select_i64_or_cmp_i32:
|
|
; GCN-DAG: s_cmp_lg_u32
|
|
; GCN: s_cselect_b64 [[CMP1:s\[[0-9]+:[0-9]+\]]], -1, 0
|
|
; GCN-DAG: s_cmp_lg_u32
|
|
; GCN: s_cselect_b64 [[CMP2:s\[[0-9]+:[0-9]+\]]], -1, 0
|
|
; GCN: s_or_b64 [[OR:s\[[0-9]+:[0-9]+\]]], [[CMP1]], [[CMP2]]
|
|
; GCN: s_and_b64 [[AND:s\[[0-9]+:[0-9]+\]]], [[OR]], exec
|
|
; GCN-DAG: s_cselect_b32 [[RESULT0:s[0-9]+]]
|
|
; GCN-DAG: s_cselect_b32 [[RESULT1:s[0-9]+]]
|
|
; GCN-DAG: v_mov_b32_e32 v[[VRESULT1:[0-9]+]], [[RESULT0]]
|
|
; GCN-DAG: v_mov_b32_e32 v[[VRESULT0:[0-9]+]], [[RESULT1]]
|
|
; GCN: buffer_store_dwordx2 v[[[VRESULT0]]:[[VRESULT1]]]
|
|
define amdgpu_kernel void @opt_select_i64_or_cmp_i32(ptr addrspace(1) %out, i32 %a, i32 %b, i32 %c, i64 %x, i64 %y) #0 {
|
|
%icmp0 = icmp ne i32 %a, %b
|
|
%icmp1 = icmp ne i32 %a, %c
|
|
%or = or i1 %icmp0, %icmp1
|
|
%select = select i1 %or, i64 %x, i64 %y
|
|
store i64 %select, ptr addrspace(1) %out
|
|
ret void
|
|
}
|
|
|
|
; GCN-LABEL: {{^}}opt_select_i64_or_cmp_f32:
|
|
; GCN-DAG: v_cmp_lg_f32_e32 vcc,
|
|
; GCN-DAG: v_cmp_lg_f32_e64 [[CMP1:s\[[0-9]+:[0-9]+\]]]
|
|
; GCN: s_or_b64 [[OR:s\[[0-9]+:[0-9]+\]]], vcc, [[CMP1]]
|
|
; GCN: s_and_b64 [[AND:s\[[0-9]+:[0-9]+\]]], [[OR]], exec
|
|
; GCN-DAG: s_cselect_b32 [[RESULT0:s[0-9]+]]
|
|
; GCN-DAG: s_cselect_b32 [[RESULT1:s[0-9]+]]
|
|
; GCN-DAG: v_mov_b32_e32 v[[VRESULT1:[0-9]+]], [[RESULT0]]
|
|
; GCN-DAG: v_mov_b32_e32 v[[VRESULT0:[0-9]+]], [[RESULT1]]
|
|
; GCN: buffer_store_dwordx2 v[[[VRESULT0]]:[[VRESULT1]]]
|
|
define amdgpu_kernel void @opt_select_i64_or_cmp_f32(ptr addrspace(1) %out, float %a, float %b, float %c, i64 %x, i64 %y) #0 {
|
|
%fcmp0 = fcmp one float %a, %b
|
|
%fcmp1 = fcmp one float %a, %c
|
|
%or = or i1 %fcmp0, %fcmp1
|
|
%select = select i1 %or, i64 %x, i64 %y
|
|
store i64 %select, ptr addrspace(1) %out
|
|
ret void
|
|
}
|
|
|
|
; GCN-LABEL: {{^}}regression:
|
|
; GCN: v_cmp_neq_f32_e64 s{{\[[0-9]+:[0-9]+\]}}, s{{[0-9]+}}, 1.0
|
|
|
|
define amdgpu_kernel void @regression(ptr addrspace(1) %out, float %c0, float %c1) #0 {
|
|
entry:
|
|
%cmp0 = fcmp oeq float %c0, 1.0
|
|
br i1 %cmp0, label %if0, label %endif
|
|
|
|
if0:
|
|
%cmp1 = fcmp oeq float %c1, 0.0
|
|
br i1 %cmp1, label %if1, label %endif
|
|
|
|
if1:
|
|
%cmp2 = xor i1 %cmp1, true
|
|
br label %endif
|
|
|
|
endif:
|
|
%tmp0 = phi i1 [ true, %entry ], [ %cmp2, %if1 ], [ false, %if0 ]
|
|
%tmp2 = select i1 %tmp0, float 4.0, float 0.0
|
|
store float %tmp2, ptr addrspace(1) %out
|
|
ret void
|
|
}
|
|
|
|
attributes #0 = { nounwind }
|