This is an archive of the discontinued LLVM Phabricator instance.

Differential D159340

[VectorCombine][NFC][test] Supplement tests of the load-extractelement sequence
Needs ReviewPublic

Authored by benshi001 on Sep 1 2023, 12:48 AM.

Details

Reviewers
spatel
dmgreen
nikic
RKSimon
fhahn
MaskRay
Summary

The newly added tests are all about scalable vector types.

Diff Detail

Revision Contents

PathSize
llvm/
test/
Transforms/
VectorCombine/
AArch64/
35 lines

Diff 555290

llvm/test/Transforms/VectorCombine/AArch64/load-extractelement-scalarization.ll

Context not available.
ret i32 %r ret i32 %r
} }
define i32 @load_extract_idx_0_vscale(ptr %x) {
; CHECK-LABEL: @load_extract_idx_0_vscale(
; CHECK-NEXT: [[LV:%.*]] = load <vscale x 4 x i32>, ptr [[X:%.*]], align 16
; CHECK-NEXT: [[R:%.*]] = extractelement <vscale x 4 x i32> [[LV]], i32 3
; CHECK-NEXT: ret i32 [[R]]
;
%lv = load <vscale x 4 x i32>, ptr %x
%r = extractelement <vscale x 4 x i32> %lv, i32 3
ret i32 %r
}
; If the original load had a smaller alignment than the scalar type, the ; If the original load had a smaller alignment than the scalar type, the
; smaller alignment should be used. ; smaller alignment should be used.
define i32 @load_extract_idx_0_small_alignment(ptr %x) { define i32 @load_extract_idx_0_small_alignment(ptr %x) {
Context not available.
ret i32 %r ret i32 %r
} }
define i32 @load_extract_idx_2_vscale(ptr %x) {
; CHECK-LABEL: @load_extract_idx_2_vscale(
; CHECK-NEXT: [[LV:%.*]] = load <vscale x 4 x i32>, ptr [[X:%.*]], align 16
; CHECK-NEXT: [[R:%.*]] = extractelement <vscale x 4 x i32> [[LV]], i32 2
; CHECK-NEXT: ret i32 [[R]]
;
%lv = load <vscale x 4 x i32>, ptr %x
%r = extractelement <vscale x 4 x i32> %lv, i32 2
ret i32 %r
}
define i32 @load_extract_idx_3(ptr %x) { define i32 @load_extract_idx_3(ptr %x) {
; CHECK-LABEL: @load_extract_idx_3( ; CHECK-LABEL: @load_extract_idx_3(
; CHECK-NEXT: [[TMP1:%.*]] = getelementptr inbounds <4 x i32>, ptr [[X:%.*]], i32 0, i32 3 ; CHECK-NEXT: [[TMP1:%.*]] = getelementptr inbounds <4 x i32>, ptr [[X:%.*]], i32 0, i32 3
Context not available.
ret i32 %r ret i32 %r
} }
; This is not optimized due to the index value 4 exceeds the minimum number
; of elements.
define i32 @load_extract_idx_4_vscale(ptr %x) {
; CHECK-LABEL: @load_extract_idx_4_vscale(
; CHECK-NEXT: [[LV:%.*]] = load <vscale x 4 x i32>, ptr [[X:%.*]], align 16
; CHECK-NEXT: [[R:%.*]] = extractelement <vscale x 4 x i32> [[LV]], i32 4
; CHECK-NEXT: ret i32 [[R]]
;
%lv = load <vscale x 4 x i32>, ptr %x
%r = extractelement <vscale x 4 x i32> %lv, i32 4
ret i32 %r
}
define i32 @load_extract_idx_var_i64(ptr %x, i64 %idx) { define i32 @load_extract_idx_var_i64(ptr %x, i64 %idx) {
; CHECK-LABEL: @load_extract_idx_var_i64( ; CHECK-LABEL: @load_extract_idx_var_i64(
; CHECK-NEXT: [[LV:%.*]] = load <4 x i32>, ptr [[X:%.*]], align 16 ; CHECK-NEXT: [[LV:%.*]] = load <4 x i32>, ptr [[X:%.*]], align 16
Context not available.