xref: /aosp_15_r20/external/llvm/test/CodeGen/X86/combine-testm-and.ll (revision 9880d6810fe72a1726cb53787c6711e909410d58)
1*9880d681SAndroid Build Coastguard Worker; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2*9880d681SAndroid Build Coastguard Worker;RUN: llc -mtriple=x86_64-apple-darwin -mcpu=skx < %s | FileCheck %s
3*9880d681SAndroid Build Coastguard Worker
4*9880d681SAndroid Build Coastguard Workerdefine i32 @combineTESTM_AND_1(<8 x i64> %a, <8 x i64> %b) {
5*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: combineTESTM_AND_1:
6*9880d681SAndroid Build Coastguard Worker; CHECK:       ## BB#0:
7*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    vptestmq %zmm0, %zmm1, %k0
8*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    kmovb %k0, %eax
9*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    retq
10*9880d681SAndroid Build Coastguard Worker  %and.i = and <8 x i64> %b, %a
11*9880d681SAndroid Build Coastguard Worker  %test.i = tail call i8 @llvm.x86.avx512.ptestm.q.512(<8 x i64> %and.i, <8 x i64> %and.i, i8 -1)
12*9880d681SAndroid Build Coastguard Worker  %conv = zext i8 %test.i to i32
13*9880d681SAndroid Build Coastguard Worker  ret i32 %conv
14*9880d681SAndroid Build Coastguard Worker}
15*9880d681SAndroid Build Coastguard Worker
16*9880d681SAndroid Build Coastguard Workerdefine i32 @combineTESTM_AND_2(<8 x i64> %a, <8 x i64> %b , i8 %mask) {
17*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: combineTESTM_AND_2:
18*9880d681SAndroid Build Coastguard Worker; CHECK:       ## BB#0:
19*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    kmovb %edi, %k1
20*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    vptestmq %zmm0, %zmm1, %k0 {%k1}
21*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    kmovb %k0, %eax
22*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    retq
23*9880d681SAndroid Build Coastguard Worker  %and.i = and <8 x i64> %b, %a
24*9880d681SAndroid Build Coastguard Worker  %test.i = tail call i8 @llvm.x86.avx512.ptestm.q.512(<8 x i64> %and.i, <8 x i64> %and.i, i8 %mask)
25*9880d681SAndroid Build Coastguard Worker  %conv = zext i8 %test.i to i32
26*9880d681SAndroid Build Coastguard Worker  ret i32 %conv
27*9880d681SAndroid Build Coastguard Worker}
28*9880d681SAndroid Build Coastguard Worker
29*9880d681SAndroid Build Coastguard Workerdefine i32 @combineTESTM_AND_mask_3(<8 x i64> %a, <8 x i64>* %bptr , i8 %mask) {
30*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: combineTESTM_AND_mask_3:
31*9880d681SAndroid Build Coastguard Worker; CHECK:       ## BB#0:
32*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    kmovb %esi, %k1
33*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    vptestmq (%rdi), %zmm0, %k0 {%k1}
34*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    kmovb %k0, %eax
35*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    retq
36*9880d681SAndroid Build Coastguard Worker  %b = load <8 x i64>, <8 x i64>* %bptr
37*9880d681SAndroid Build Coastguard Worker  %and.i = and <8 x i64> %a, %b
38*9880d681SAndroid Build Coastguard Worker  %test.i = tail call i8 @llvm.x86.avx512.ptestm.q.512(<8 x i64> %and.i, <8 x i64> %and.i, i8 %mask)
39*9880d681SAndroid Build Coastguard Worker  %conv = zext i8 %test.i to i32
40*9880d681SAndroid Build Coastguard Worker  ret i32 %conv
41*9880d681SAndroid Build Coastguard Worker}
42*9880d681SAndroid Build Coastguard Worker
43*9880d681SAndroid Build Coastguard Workerdefine i32 @combineTESTM_AND_mask_4(<8 x i64> %a, <8 x i64>* %bptr , i8 %mask) {
44*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: combineTESTM_AND_mask_4:
45*9880d681SAndroid Build Coastguard Worker; CHECK:       ## BB#0:
46*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    kmovb %esi, %k1
47*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    vptestmq (%rdi), %zmm0, %k0 {%k1}
48*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    kmovb %k0, %eax
49*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:    retq
50*9880d681SAndroid Build Coastguard Worker  %b = load <8 x i64>, <8 x i64>* %bptr
51*9880d681SAndroid Build Coastguard Worker  %and.i = and <8 x i64> %b, %a
52*9880d681SAndroid Build Coastguard Worker  %test.i = tail call i8 @llvm.x86.avx512.ptestm.q.512(<8 x i64> %and.i, <8 x i64> %and.i, i8 %mask)
53*9880d681SAndroid Build Coastguard Worker  %conv = zext i8 %test.i to i32
54*9880d681SAndroid Build Coastguard Worker  ret i32 %conv
55*9880d681SAndroid Build Coastguard Worker}
56*9880d681SAndroid Build Coastguard Worker
57*9880d681SAndroid Build Coastguard Workerdeclare i8 @llvm.x86.avx512.ptestm.q.512(<8 x i64>, <8 x i64>, i8)
58