File: vec_extract_through_broadcast.ll

package info (click to toggle)
llvm-toolchain-snapshot 1%3A22~%2B%2B20250731080150%2Bbe449d6b6587-1~exp1
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 2,250,472 kB
  • sloc: cpp: 7,641,756; ansic: 1,439,220; asm: 1,072,591; python: 253,178; f90: 95,173; objc: 70,722; lisp: 44,365; pascal: 18,720; sh: 10,051; ml: 5,111; perl: 4,720; awk: 3,523; makefile: 3,397; javascript: 2,272; xml: 892; fortran: 783
file content (52 lines) | stat: -rw-r--r-- 2,178 bytes parent folder | download | duplicates (7)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
; RUN: opt -passes=instcombine -S < %s | FileCheck %s

define float @extract_from_zero_init_shuffle(<2 x float> %vec, i64 %idx) {
; CHECK-LABEL: @extract_from_zero_init_shuffle(
; CHECK-NEXT:    %extract = extractelement <2 x float> %vec, i64 0
; CHECK-NEXT:    ret float %extract
;
  %shuffle = shufflevector <2 x float> %vec, <2 x float> poison, <4 x i32> zeroinitializer
  %extract = extractelement <4 x float> %shuffle, i64 %idx
  ret float %extract
}


define float @extract_from_general_splat(<2 x float> %vec, i64 %idx) {
; CHECK-LABEL: @extract_from_general_splat(
; CHECK-NEXT:    %extract = extractelement <2 x float> %vec, i64 1
; CHECK-NEXT:    ret float %extract
;
  %shuffle = shufflevector <2 x float> %vec, <2 x float> poison, <4 x i32> <i32 1, i32 1, i32 1, i32 1>
  %extract = extractelement <4 x float> %shuffle, i64 %idx
  ret float %extract
}

define float @extract_from_general_scalable_splat(<vscale x 2 x float> %vec, i64 %idx) {
; CHECK-LABEL: @extract_from_general_scalable_splat(
; CHECK-NEXT:    %extract = extractelement <vscale x 2 x float> %vec, i64 0
; CHECK-NEXT:    ret float %extract
;
  %shuffle = shufflevector <vscale x 2 x float> %vec, <vscale x 2 x float> poison, <vscale x 4 x i32> zeroinitializer
  %extract = extractelement <vscale x 4 x float> %shuffle, i64 %idx
  ret float %extract
}

define float @extract_from_splat_with_poison_0(<2 x float> %vec, i64 %idx) {
; CHECK-LABEL: @extract_from_splat_with_poison_0(
; CHECK-NEXT:    %extract = extractelement <2 x float> %vec, i64 1
; CHECK-NEXT:    ret float %extract
;
  %shuffle = shufflevector <2 x float> %vec, <2 x float> poison, <4 x i32> <i32 poison, i32 1, i32 1, i32 1>
  %extract = extractelement <4 x float> %shuffle, i64 %idx
  ret float %extract
}

define float @extract_from_splat_with_poison_1(<2 x float> %vec, i64 %idx) {
; CHECK-LABEL: @extract_from_splat_with_poison_1(
; CHECK-NEXT:    %extract = extractelement <2 x float> %vec, i64 1
; CHECK-NEXT:    ret float %extract
;
  %shuffle = shufflevector <2 x float> %vec, <2 x float> poison, <4 x i32> <i32 1, i32 poison, i32 1, i32 1>
  %extract = extractelement <4 x float> %shuffle, i64 %idx
  ret float %extract
}