derek/TheRock
1
0 Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
1a182e2dac58879f215c0fad0f5f78c76dcafe56
TheRock/compiler/amd-llvm/llvm/test/Transforms/InstCombine/select-gep.ll
T
derek.holloway 7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
Details
CodeQL / Analyze (python) (push) Canceled after 0s
Details
pre-commit / pre-commit (push) Canceled after 0s
Details
Multi-Arch CI / setup (push) Canceled after 0s
Details
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Details
Setup (Multi-Arch) / setup (push) Canceled after 0s
Details
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Details
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
Details
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Details
Multi-Arch CI / CI Summary (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Details
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Details
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
Details
Flatten subprojects into a single base monorepo
2026-09-02 16:34:47 -07:00

320 lines
13 KiB
LLVM
Raw Blame History

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt < %s -passes=instcombine -S | FileCheck %s
define ptr @test1a(ptr %p, ptr %q) {
; CHECK-LABEL: @test1a(
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt ptr [[P:%.*]], [[Q:%.*]]
; CHECK-NEXT: [[SELECT_V:%.*]] = select i1 [[CMP]], ptr [[P]], ptr [[Q]]
; CHECK-NEXT: [[SELECT:%.*]] = getelementptr i8, ptr [[SELECT_V]], i64 16
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep1 = getelementptr i32, ptr %p, i64 4
%gep2 = getelementptr i32, ptr %q, i64 4
%cmp = icmp ugt ptr %p, %q
%select = select i1 %cmp, ptr %gep1, ptr %gep2
ret ptr %select
}
define ptr @test1b(ptr %p, ptr %q) {
; CHECK-LABEL: @test1b(
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt ptr [[P:%.*]], [[Q:%.*]]
; CHECK-NEXT: [[SELECT_V:%.*]] = select i1 [[CMP]], ptr [[P]], ptr [[Q]]
; CHECK-NEXT: [[SELECT:%.*]] = getelementptr i8, ptr [[SELECT_V]], i64 16
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep1 = getelementptr inbounds i32, ptr %p, i64 4
%gep2 = getelementptr i32, ptr %q, i64 4
%cmp = icmp ugt ptr %p, %q
%select = select i1 %cmp, ptr %gep1, ptr %gep2
ret ptr %select
}
define ptr @test1c(ptr %p, ptr %q) {
; CHECK-LABEL: @test1c(
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt ptr [[P:%.*]], [[Q:%.*]]
; CHECK-NEXT: [[SELECT_V:%.*]] = select i1 [[CMP]], ptr [[P]], ptr [[Q]]
; CHECK-NEXT: [[SELECT:%.*]] = getelementptr i8, ptr [[SELECT_V]], i64 16
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep1 = getelementptr i32, ptr %p, i64 4
%gep2 = getelementptr inbounds i32, ptr %q, i64 4
%cmp = icmp ugt ptr %p, %q
%select = select i1 %cmp, ptr %gep1, ptr %gep2
ret ptr %select
}
define ptr @test1d(ptr %p, ptr %q) {
; CHECK-LABEL: @test1d(
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt ptr [[P:%.*]], [[Q:%.*]]
; CHECK-NEXT: [[SELECT_V:%.*]] = select i1 [[CMP]], ptr [[P]], ptr [[Q]]
; CHECK-NEXT: [[SELECT:%.*]] = getelementptr inbounds nuw i8, ptr [[SELECT_V]], i64 16
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep1 = getelementptr inbounds i32, ptr %p, i64 4
%gep2 = getelementptr inbounds i32, ptr %q, i64 4
%cmp = icmp ugt ptr %p, %q
%select = select i1 %cmp, ptr %gep1, ptr %gep2
ret ptr %select
}
define ptr @test2(ptr %p, i64 %x, i64 %y) {
; CHECK-LABEL: @test2(
; CHECK-NEXT: [[SELECT_V:%.*]] = call i64 @llvm.umax.i64(i64 [[X:%.*]], i64 [[Y:%.*]])
; CHECK-NEXT: [[SELECT:%.*]] = getelementptr inbounds [4 x i8], ptr [[P:%.*]], i64 [[SELECT_V]]
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep1 = getelementptr inbounds i32, ptr %p, i64 %x
%gep2 = getelementptr inbounds i32, ptr %p, i64 %y
%cmp = icmp ugt i64 %x, %y
%select = select i1 %cmp, ptr %gep1, ptr %gep2
ret ptr %select
}
; PR50183
define ptr @test2a(ptr %p, i64 %x, i64 %y) {
; CHECK-LABEL: @test2a(
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt i64 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[SELECT_IDX:%.*]] = select i1 [[CMP]], i64 [[X]], i64 0
; CHECK-NEXT: [[SELECT:%.*]] = getelementptr inbounds [4 x i8], ptr [[P:%.*]], i64 [[SELECT_IDX]]
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep = getelementptr inbounds i32, ptr %p, i64 %x
%cmp = icmp ugt i64 %x, %y
%select = select i1 %cmp, ptr %gep, ptr %p
ret ptr %select
}
define ptr @test2a_nusw(ptr %p, i64 %x, i64 %y) {
; CHECK-LABEL: @test2a_nusw(
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt i64 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[SELECT_IDX:%.*]] = select i1 [[CMP]], i64 [[X]], i64 0
; CHECK-NEXT: [[SELECT:%.*]] = getelementptr nusw [4 x i8], ptr [[P:%.*]], i64 [[SELECT_IDX]]
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep = getelementptr nusw i32, ptr %p, i64 %x
%cmp = icmp ugt i64 %x, %y
%select = select i1 %cmp, ptr %gep, ptr %p
ret ptr %select
}
define ptr @test2a_nuw(ptr %p, i64 %x, i64 %y) {
; CHECK-LABEL: @test2a_nuw(
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt i64 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[SELECT_IDX:%.*]] = select i1 [[CMP]], i64 [[X]], i64 0
; CHECK-NEXT: [[SELECT:%.*]] = getelementptr nuw [4 x i8], ptr [[P:%.*]], i64 [[SELECT_IDX]]
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep = getelementptr nuw i32, ptr %p, i64 %x
%cmp = icmp ugt i64 %x, %y
%select = select i1 %cmp, ptr %gep, ptr %p
ret ptr %select
}
; PR50183
define ptr @test2b(ptr %p, i64 %x, i64 %y) {
; CHECK-LABEL: @test2b(
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt i64 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[SELECT_IDX:%.*]] = select i1 [[CMP]], i64 0, i64 [[X]]
; CHECK-NEXT: [[SELECT:%.*]] = getelementptr inbounds [4 x i8], ptr [[P:%.*]], i64 [[SELECT_IDX]]
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep = getelementptr inbounds i32, ptr %p, i64 %x
%cmp = icmp ugt i64 %x, %y
%select = select i1 %cmp, ptr %p, ptr %gep
ret ptr %select
}
; PR51069
define ptr @test2c(ptr %p, i64 %x, i64 %y) {
; CHECK-LABEL: @test2c(
; CHECK-NEXT: [[GEP1:%.*]] = getelementptr inbounds [4 x i8], ptr [[P:%.*]], i64 [[X:%.*]]
; CHECK-NEXT: [[ICMP:%.*]] = icmp ugt i64 [[X]], [[Y:%.*]]
; CHECK-NEXT: [[SEL_IDX:%.*]] = select i1 [[ICMP]], i64 0, i64 24
; CHECK-NEXT: [[SEL:%.*]] = getelementptr inbounds nuw i8, ptr [[GEP1]], i64 [[SEL_IDX]]
; CHECK-NEXT: ret ptr [[SEL]]
;
%gep1 = getelementptr inbounds i32, ptr %p, i64 %x
%gep2 = getelementptr inbounds i32, ptr %gep1, i64 6
%icmp = icmp ugt i64 %x, %y
%sel = select i1 %icmp, ptr %gep1, ptr %gep2
ret ptr %sel
}
; PR51069
define ptr @test2d(ptr %p, i64 %x, i64 %y) {
; CHECK-LABEL: @test2d(
; CHECK-NEXT: [[GEP1:%.*]] = getelementptr inbounds [4 x i8], ptr [[P:%.*]], i64 [[X:%.*]]
; CHECK-NEXT: [[ICMP:%.*]] = icmp ugt i64 [[X]], [[Y:%.*]]
; CHECK-NEXT: [[SEL_IDX:%.*]] = select i1 [[ICMP]], i64 24, i64 0
; CHECK-NEXT: [[SEL:%.*]] = getelementptr inbounds nuw i8, ptr [[GEP1]], i64 [[SEL_IDX]]
; CHECK-NEXT: ret ptr [[SEL]]
;
%gep1 = getelementptr inbounds i32, ptr %p, i64 %x
%gep2 = getelementptr inbounds i32, ptr %gep1, i64 6
%icmp = icmp ugt i64 %x, %y
%sel = select i1 %icmp, ptr %gep2, ptr %gep1
ret ptr %sel
}
; Three (or more) operand GEPs are currently expected to not be optimised,
; though they could be in principle.
define ptr @test3a(ptr %p, i64 %x, i64 %y) {
; CHECK-LABEL: @test3a(
; CHECK-NEXT: [[GEP1_SPLIT:%.*]] = getelementptr inbounds nuw i8, ptr [[P:%.*]], i64 32
; CHECK-NEXT: [[GEP1:%.*]] = getelementptr inbounds [4 x i8], ptr [[GEP1_SPLIT]], i64 [[X:%.*]]
; CHECK-NEXT: [[GEP2_SPLIT:%.*]] = getelementptr inbounds nuw i8, ptr [[P]], i64 32
; CHECK-NEXT: [[GEP2:%.*]] = getelementptr inbounds [4 x i8], ptr [[GEP2_SPLIT]], i64 [[Y:%.*]]
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt i64 [[X]], [[Y]]
; CHECK-NEXT: [[SELECT:%.*]] = select i1 [[CMP]], ptr [[GEP1]], ptr [[GEP2]]
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep1 = getelementptr inbounds [4 x i32], ptr %p, i64 2, i64 %x
%gep2 = getelementptr inbounds [4 x i32], ptr %p, i64 2, i64 %y
%cmp = icmp ugt i64 %x, %y
%select = select i1 %cmp, ptr %gep1, ptr %gep2
ret ptr %select
}
define ptr @test3b(ptr %p, ptr %q, i64 %x, i64 %y) {
; CHECK-LABEL: @test3b(
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt i64 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[SELECT:%.*]] = select i1 [[CMP]], ptr [[GEP1:%.*]], ptr [[GEP2:%.*]]
; CHECK-NEXT: [[SELECT_V:%.*]] = getelementptr inbounds nuw i8, ptr [[SELECT]], i64 32
; CHECK-NEXT: [[SELECT1:%.*]] = getelementptr inbounds [4 x i8], ptr [[SELECT_V]], i64 [[X]]
; CHECK-NEXT: ret ptr [[SELECT1]]
;
%gep1 = getelementptr inbounds [4 x i32], ptr %p, i64 2, i64 %x
%gep2 = getelementptr inbounds [4 x i32], ptr %q, i64 2, i64 %x
%cmp = icmp ugt i64 %x, %y
%select = select i1 %cmp, ptr %gep1, ptr %gep2
ret ptr %select
}
define ptr @test3c(ptr %p, ptr %q, i64 %x, i64 %y) {
; CHECK-LABEL: @test3c(
; CHECK-NEXT: [[GEP1_SPLIT:%.*]] = getelementptr inbounds [16 x i8], ptr [[P:%.*]], i64 [[X:%.*]]
; CHECK-NEXT: [[GEP1:%.*]] = getelementptr inbounds nuw i8, ptr [[GEP1_SPLIT]], i64 8
; CHECK-NEXT: [[GEP2:%.*]] = getelementptr inbounds [4 x i8], ptr [[Q:%.*]], i64 [[X]]
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt i64 [[X]], [[Y:%.*]]
; CHECK-NEXT: [[SELECT:%.*]] = select i1 [[CMP]], ptr [[GEP1]], ptr [[GEP2]]
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep1 = getelementptr inbounds [4 x i32], ptr %p, i64 %x, i64 2
%gep2 = getelementptr inbounds i32, ptr %q, i64 %x
%cmp = icmp ugt i64 %x, %y
%select = select i1 %cmp, ptr %gep1, ptr %gep2
ret ptr %select
}
define ptr @test3d(ptr %p, ptr %q, i64 %x, i64 %y) {
; CHECK-LABEL: @test3d(
; CHECK-NEXT: [[GEP1:%.*]] = getelementptr inbounds [4 x i8], ptr [[P:%.*]], i64 [[X:%.*]]
; CHECK-NEXT: [[GEP2_SPLIT:%.*]] = getelementptr inbounds [16 x i8], ptr [[Q:%.*]], i64 [[X]]
; CHECK-NEXT: [[GEP2:%.*]] = getelementptr inbounds nuw i8, ptr [[GEP2_SPLIT]], i64 8
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt i64 [[X]], [[Y:%.*]]
; CHECK-NEXT: [[SELECT:%.*]] = select i1 [[CMP]], ptr [[GEP1]], ptr [[GEP2]]
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep1 = getelementptr inbounds i32, ptr %p, i64 %x
%gep2 = getelementptr inbounds [4 x i32], ptr %q, i64 %x, i64 2
%cmp = icmp ugt i64 %x, %y
%select = select i1 %cmp, ptr %gep1, ptr %gep2
ret ptr %select
}
; Shouldn't be optimised as it would mean introducing an extra select
define ptr @test4(ptr %p, ptr %q, i64 %x, i64 %y) {
; CHECK-LABEL: @test4(
; CHECK-NEXT: [[GEP1:%.*]] = getelementptr inbounds [4 x i8], ptr [[P:%.*]], i64 [[X:%.*]]
; CHECK-NEXT: [[GEP2:%.*]] = getelementptr inbounds [4 x i8], ptr [[Q:%.*]], i64 [[Y:%.*]]
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt i64 [[X]], [[Y]]
; CHECK-NEXT: [[SELECT:%.*]] = select i1 [[CMP]], ptr [[GEP1]], ptr [[GEP2]]
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep1 = getelementptr inbounds i32, ptr %p, i64 %x
%gep2 = getelementptr inbounds i32, ptr %q, i64 %y
%cmp = icmp ugt i64 %x, %y
%select = select i1 %cmp, ptr %gep1, ptr %gep2
ret ptr %select
}
; We cannot create a select with a vector condition but scalar operands.
define <2 x ptr> @test5(ptr %p1, ptr %p2, <2 x i64> %idx, <2 x i1> %cc) {
; CHECK-LABEL: @test5(
; CHECK-NEXT: [[GEP1:%.*]] = getelementptr [8 x i8], ptr [[P1:%.*]], <2 x i64> [[IDX:%.*]]
; CHECK-NEXT: [[GEP2:%.*]] = getelementptr [8 x i8], ptr [[P2:%.*]], <2 x i64> [[IDX]]
; CHECK-NEXT: [[SELECT:%.*]] = select <2 x i1> [[CC:%.*]], <2 x ptr> [[GEP1]], <2 x ptr> [[GEP2]]
; CHECK-NEXT: ret <2 x ptr> [[SELECT]]
;
%gep1 = getelementptr i64, ptr %p1, <2 x i64> %idx
%gep2 = getelementptr i64, ptr %p2, <2 x i64> %idx
%select = select <2 x i1> %cc, <2 x ptr> %gep1, <2 x ptr> %gep2
ret <2 x ptr> %select
}
; PR51069 - multiple uses
define ptr @test6(ptr %p, i64 %x, i64 %y) {
; CHECK-LABEL: @test6(
; CHECK-NEXT: [[GEP1:%.*]] = getelementptr inbounds [4 x i8], ptr [[P:%.*]], i64 [[X:%.*]]
; CHECK-NEXT: [[ICMP:%.*]] = icmp ugt i64 [[X]], [[Y:%.*]]
; CHECK-NEXT: [[SEL_IDX:%.*]] = select i1 [[ICMP]], i64 [[Y]], i64 0
; CHECK-NEXT: [[SEL:%.*]] = getelementptr inbounds [4 x i8], ptr [[GEP1]], i64 [[SEL_IDX]]
; CHECK-NEXT: call void @use_i32p(ptr [[GEP1]])
; CHECK-NEXT: ret ptr [[SEL]]
;
%gep1 = getelementptr inbounds i32, ptr %p, i64 %x
%gep2 = getelementptr inbounds i32, ptr %gep1, i64 %y
%icmp = icmp ugt i64 %x, %y
%sel = select i1 %icmp, ptr %gep2, ptr %gep1
call void @use_i32p(ptr %gep1)
ret ptr %sel
}
declare void @use_i32p(ptr)
; We cannot create a select-with-idx with a vector condition but scalar idx.
define <2 x ptr> @test7(<2 x ptr> %p1, i64 %idx, <2 x i1> %cc) {
; CHECK-LABEL: @test7(
; CHECK-NEXT: [[GEP:%.*]] = getelementptr [8 x i8], <2 x ptr> [[P1:%.*]], i64 [[IDX:%.*]]
; CHECK-NEXT: [[SELECT:%.*]] = select <2 x i1> [[CC:%.*]], <2 x ptr> [[P1]], <2 x ptr> [[GEP]]
; CHECK-NEXT: ret <2 x ptr> [[SELECT]]
;
%gep = getelementptr i64, <2 x ptr> %p1, i64 %idx
%select = select <2 x i1> %cc, <2 x ptr> %p1, <2 x ptr> %gep
ret <2 x ptr> %select
}
define ptr @ptr_eq_replace_freeze1(ptr %p, ptr %q) {
; CHECK-LABEL: @ptr_eq_replace_freeze1(
; CHECK-NEXT: [[Q_FR:%.*]] = freeze ptr [[Q:%.*]]
; CHECK-NEXT: [[Q_FR1:%.*]] = freeze ptr [[Q1:%.*]]
; CHECK-NEXT: [[CMP:%.*]] = icmp eq ptr [[Q_FR]], [[Q_FR1]]
; CHECK-NEXT: [[SELECT:%.*]] = select i1 [[CMP]], ptr [[Q_FR]], ptr [[Q_FR1]]
; CHECK-NEXT: ret ptr [[SELECT]]
;
%p.fr = freeze ptr %p
%q.fr = freeze ptr %q
%cmp = icmp eq ptr %p.fr, %q.fr
%select = select i1 %cmp, ptr %p.fr, ptr %q.fr
ret ptr %select
}
define ptr @ptr_eq_replace_freeze2(ptr %p, ptr %q) {
; CHECK-LABEL: @ptr_eq_replace_freeze2(
; CHECK-NEXT: [[CMP:%.*]] = icmp eq ptr [[P_FR1:%.*]], [[P_FR:%.*]]
; CHECK-NEXT: [[CMP_FR:%.*]] = freeze i1 [[CMP]]
; CHECK-NEXT: [[SELECT_V:%.*]] = select i1 [[CMP_FR]], ptr [[P_FR1]], ptr [[P_FR]]
; CHECK-NEXT: [[SELECT:%.*]] = getelementptr i8, ptr [[SELECT_V]], i64 16
; CHECK-NEXT: ret ptr [[SELECT]]
;
%gep1 = getelementptr i32, ptr %p, i64 4
%gep2 = getelementptr i32, ptr %q, i64 4
%cmp = icmp eq ptr %p, %q
%cmp.fr = freeze i1 %cmp
%select = select i1 %cmp.fr, ptr %gep1, ptr %gep2
ret ptr %select
}
Reference in new issue View Git Blame Copy Permalink