llvm/test/CodeGen/X86/bitselect.ll

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; RUN: llc < %s -mtriple=i686-unknown-unknown | FileCheck %s --check-prefixes=X86
; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=-bmi | FileCheck %s --check-prefixes=X64-NOBMI
; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+bmi | FileCheck %s --check-prefixes=X64-BMI

; PR46472
; bitselect(a,b,m) == or(and(a,not(m)),and(b,m))
; bitselect(a,b,m) == xor(and(xor(a,b),m),a)

define i8 @bitselect_i8(i8 %a, i8 %b, i8 %m) nounwind {
; X86-LABEL: bitselect_i8:
; X86:       # %bb.0:
; X86-NEXT:    movzbl {{[0-9]+}}(%esp), %ecx
; X86-NEXT:    movzbl {{[0-9]+}}(%esp), %eax
; X86-NEXT:    xorb %cl, %al
; X86-NEXT:    andb {{[0-9]+}}(%esp), %al
; X86-NEXT:    xorb %cl, %al
; X86-NEXT:    retl
;
; X64-NOBMI-LABEL: bitselect_i8:
; X64-NOBMI:       # %bb.0:
; X64-NOBMI-NEXT:    movl %esi, %eax
; X64-NOBMI-NEXT:    xorl %edi, %eax
; X64-NOBMI-NEXT:    andl %edx, %eax
; X64-NOBMI-NEXT:    xorl %edi, %eax
; X64-NOBMI-NEXT:    # kill: def $al killed $al killed $eax
; X64-NOBMI-NEXT:    retq
;
; X64-BMI-LABEL: bitselect_i8:
; X64-BMI:       # %bb.0:
; X64-BMI-NEXT:    andnl %edi, %edx, %eax
; X64-BMI-NEXT:    andl %edx, %esi
; X64-BMI-NEXT:    orl %esi, %eax
; X64-BMI-NEXT:    # kill: def $al killed $al killed $eax
; X64-BMI-NEXT:    retq
  %not = xor i8 %m, -1
  %ma = and i8 %a, %not
  %mb = and i8 %b, %m
  %or = or i8 %ma, %mb
  ret i8 %or
}

define i16 @bitselect_i16(i16 %a, i16 %b, i16 %m) nounwind {
; X86-LABEL: bitselect_i16:
; X86:       # %bb.0:
; X86-NEXT:    movl {{[0-9]+}}(%esp), %ecx
; X86-NEXT:    movzwl {{[0-9]+}}(%esp), %eax
; X86-NEXT:    xorw %cx, %ax
; X86-NEXT:    andw {{[0-9]+}}(%esp), %ax
; X86-NEXT:    xorl %ecx, %eax
; X86-NEXT:    # kill: def $ax killed $ax killed $eax
; X86-NEXT:    retl
;
; X64-NOBMI-LABEL: bitselect_i16:
; X64-NOBMI:       # %bb.0:
; X64-NOBMI-NEXT:    movl %esi, %eax
; X64-NOBMI-NEXT:    xorl %edi, %eax
; X64-NOBMI-NEXT:    andl %edx, %eax
; X64-NOBMI-NEXT:    xorl %edi, %eax
; X64-NOBMI-NEXT:    # kill: def $ax killed $ax killed $eax
; X64-NOBMI-NEXT:    retq
;
; X64-BMI-LABEL: bitselect_i16:
; X64-BMI:       # %bb.0:
; X64-BMI-NEXT:    andnl %edi, %edx, %eax
; X64-BMI-NEXT:    andl %edx, %esi
; X64-BMI-NEXT:    orl %esi, %eax
; X64-BMI-NEXT:    # kill: def $ax killed $ax killed $eax
; X64-BMI-NEXT:    retq
  %not = xor i16 %m, -1
  %ma = and i16 %a, %not
  %mb = and i16 %b, %m
  %or = or i16 %ma, %mb
  ret i16 %or
}

define i32 @bitselect_i32(i32 %a, i32 %b, i32 %m) nounwind {
; X86-LABEL: bitselect_i32:
; X86:       # %bb.0:
; X86-NEXT:    movl {{[0-9]+}}(%esp), %ecx
; X86-NEXT:    movl {{[0-9]+}}(%esp), %eax
; X86-NEXT:    xorl %ecx, %eax
; X86-NEXT:    andl {{[0-9]+}}(%esp), %eax
; X86-NEXT:    xorl %ecx, %eax
; X86-NEXT:    retl
;
; X64-NOBMI-LABEL: bitselect_i32:
; X64-NOBMI:       # %bb.0:
; X64-NOBMI-NEXT:    movl %esi, %eax
; X64-NOBMI-NEXT:    xorl %edi, %eax
; X64-NOBMI-NEXT:    andl %edx, %eax
; X64-NOBMI-NEXT:    xorl %edi, %eax
; X64-NOBMI-NEXT:    retq
;
; X64-BMI-LABEL: bitselect_i32:
; X64-BMI:       # %bb.0:
; X64-BMI-NEXT:    andnl %edi, %edx, %eax
; X64-BMI-NEXT:    andl %edx, %esi
; X64-BMI-NEXT:    orl %esi, %eax
; X64-BMI-NEXT:    retq
  %not = xor i32 %m, -1
  %ma = and i32 %a, %not
  %mb = and i32 %b, %m
  %or = or i32 %ma, %mb
  ret i32 %or
}

define i64 @bitselect_i64(i64 %a, i64 %b, i64 %m) nounwind {
; X86-LABEL: bitselect_i64:
; X86:       # %bb.0:
; X86-NEXT:    movl {{[0-9]+}}(%esp), %edx
; X86-NEXT:    movl {{[0-9]+}}(%esp), %ecx
; X86-NEXT:    movl {{[0-9]+}}(%esp), %eax
; X86-NEXT:    xorl %edx, %eax
; X86-NEXT:    andl {{[0-9]+}}(%esp), %eax
; X86-NEXT:    xorl %edx, %eax
; X86-NEXT:    movl {{[0-9]+}}(%esp), %edx
; X86-NEXT:    xorl %ecx, %edx
; X86-NEXT:    andl {{[0-9]+}}(%esp), %edx
; X86-NEXT:    xorl %ecx, %edx
; X86-NEXT:    retl
;
; X64-NOBMI-LABEL: bitselect_i64:
; X64-NOBMI:       # %bb.0:
; X64-NOBMI-NEXT:    movq %rsi, %rax
; X64-NOBMI-NEXT:    xorq %rdi, %rax
; X64-NOBMI-NEXT:    andq %rdx, %rax
; X64-NOBMI-NEXT:    xorq %rdi, %rax
; X64-NOBMI-NEXT:    retq
;
; X64-BMI-LABEL: bitselect_i64:
; X64-BMI:       # %bb.0:
; X64-BMI-NEXT:    andnq %rdi, %rdx, %rax
; X64-BMI-NEXT:    andq %rdx, %rsi
; X64-BMI-NEXT:    orq %rsi, %rax
; X64-BMI-NEXT:    retq
  %not = xor i64 %m, -1
  %ma = and i64 %a, %not
  %mb = and i64 %b, %m
  %or = or i64 %ma, %mb
  ret i64 %or
}

define i128 @bitselect_i128(i128 %a, i128 %b, i128 %m) nounwind {
; X86-LABEL: bitselect_i128:
; X86:       # %bb.0:
; X86-NEXT:    pushl %ebp
; X86-NEXT:    movl %esp, %ebp
; X86-NEXT:    pushl %edi
; X86-NEXT:    pushl %esi
; X86-NEXT:    andl $-16, %esp
; X86-NEXT:    movl 32(%ebp), %edx
; X86-NEXT:    movl 36(%ebp), %eax
; X86-NEXT:    movl 24(%ebp), %esi
; X86-NEXT:    movl 28(%ebp), %edi
; X86-NEXT:    movl 40(%ebp), %ecx
; X86-NEXT:    xorl %esi, %ecx
; X86-NEXT:    andl 56(%ebp), %ecx
; X86-NEXT:    xorl %esi, %ecx
; X86-NEXT:    movl 44(%ebp), %esi
; X86-NEXT:    xorl %edi, %esi
; X86-NEXT:    andl 60(%ebp), %esi
; X86-NEXT:    xorl %edi, %esi
; X86-NEXT:    movl 48(%ebp), %edi
; X86-NEXT:    xorl %edx, %edi
; X86-NEXT:    andl 64(%ebp), %edi
; X86-NEXT:    xorl %edx, %edi
; X86-NEXT:    movl 52(%ebp), %edx
; X86-NEXT:    xorl %eax, %edx
; X86-NEXT:    andl 68(%ebp), %edx
; X86-NEXT:    xorl %eax, %edx
; X86-NEXT:    movl 8(%ebp), %eax
; X86-NEXT:    movl %edx, 12(%eax)
; X86-NEXT:    movl %edi, 8(%eax)
; X86-NEXT:    movl %esi, 4(%eax)
; X86-NEXT:    movl %ecx, (%eax)
; X86-NEXT:    leal -8(%ebp), %esp
; X86-NEXT:    popl %esi
; X86-NEXT:    popl %edi
; X86-NEXT:    popl %ebp
; X86-NEXT:    retl $4
;
; X64-NOBMI-LABEL: bitselect_i128:
; X64-NOBMI:       # %bb.0:
; X64-NOBMI-NEXT:    movq %rdx, %rax
; X64-NOBMI-NEXT:    xorq %rdi, %rax
; X64-NOBMI-NEXT:    andq %r8, %rax
; X64-NOBMI-NEXT:    xorq %rdi, %rax
; X64-NOBMI-NEXT:    xorq %rsi, %rcx
; X64-NOBMI-NEXT:    andq %r9, %rcx
; X64-NOBMI-NEXT:    xorq %rsi, %rcx
; X64-NOBMI-NEXT:    movq %rcx, %rdx
; X64-NOBMI-NEXT:    retq
;
; X64-BMI-LABEL: bitselect_i128:
; X64-BMI:       # %bb.0:
; X64-BMI-NEXT:    andnq %rdi, %r8, %rax
; X64-BMI-NEXT:    andq %r8, %rdx
; X64-BMI-NEXT:    orq %rdx, %rax
; X64-BMI-NEXT:    andnq %rsi, %r9, %rdx
; X64-BMI-NEXT:    andq %r9, %rcx
; X64-BMI-NEXT:    orq %rcx, %rdx
; X64-BMI-NEXT:    retq
  %not = xor i128 %m, -1
  %ma = and i128 %a, %not
  %mb = and i128 %b, %m
  %or = or i128 %ma, %mb
  ret i128 %or
}

;
; Bitselect between constants
;

; bitselect(52, -6553, m)
; TODO: Non-BMI canonicalization is actually better.
define i32 @bitselect_constants_i32(i32 %m) nounwind {
; X86-LABEL: bitselect_constants_i32:
; X86:       # %bb.0:
; X86-NEXT:    movl $-6573, %eax # imm = 0xE653
; X86-NEXT:    andl {{[0-9]+}}(%esp), %eax
; X86-NEXT:    xorl $52, %eax
; X86-NEXT:    retl
;
; X64-NOBMI-LABEL: bitselect_constants_i32:
; X64-NOBMI:       # %bb.0:
; X64-NOBMI-NEXT:    movl %edi, %eax
; X64-NOBMI-NEXT:    andl $-6573, %eax # imm = 0xE653
; X64-NOBMI-NEXT:    xorl $52, %eax
; X64-NOBMI-NEXT:    retq
;
; X64-BMI-LABEL: bitselect_constants_i32:
; X64-BMI:       # %bb.0:
; X64-BMI-NEXT:    movl %edi, %eax
; X64-BMI-NEXT:    notl %eax
; X64-BMI-NEXT:    andl $52, %eax
; X64-BMI-NEXT:    andl $-6553, %edi # imm = 0xE667
; X64-BMI-NEXT:    orl %edi, %eax
; X64-BMI-NEXT:    retq
  %not = xor i32 %m, -1
  %ma = and i32 52, %not
  %mb = and i32 -6553, %m
  %or = or i32 %ma, %mb
  ret i32 %or
}