forked from OSchip/llvm-project
143 lines
3.6 KiB
LLVM
143 lines
3.6 KiB
LLVM
; RUN: llc < %s -mtriple=x86_64-unknown-unknown | FileCheck %s
|
|
|
|
declare i8 @llvm.cttz.i8(i8, i1)
|
|
declare i16 @llvm.cttz.i16(i16, i1)
|
|
declare i32 @llvm.cttz.i32(i32, i1)
|
|
declare i64 @llvm.cttz.i64(i64, i1)
|
|
declare i8 @llvm.ctlz.i8(i8, i1)
|
|
declare i16 @llvm.ctlz.i16(i16, i1)
|
|
declare i32 @llvm.ctlz.i32(i32, i1)
|
|
declare i64 @llvm.ctlz.i64(i64, i1)
|
|
|
|
define i8 @cttz_i8(i8 %x) {
|
|
; CHECK-LABEL: cttz_i8:
|
|
; CHECK: # BB#0:
|
|
; CHECK-NEXT: movzbl %dil, %eax
|
|
; CHECK-NEXT: bsfl %eax, %eax
|
|
; CHECK-NEXT: retq
|
|
%tmp = call i8 @llvm.cttz.i8( i8 %x, i1 true )
|
|
ret i8 %tmp
|
|
}
|
|
|
|
define i16 @cttz_i16(i16 %x) {
|
|
; CHECK-LABEL: cttz_i16:
|
|
; CHECK: # BB#0:
|
|
; CHECK-NEXT: bsfw %di, %ax
|
|
; CHECK-NEXT: retq
|
|
%tmp = call i16 @llvm.cttz.i16( i16 %x, i1 true )
|
|
ret i16 %tmp
|
|
}
|
|
|
|
define i32 @cttz_i32(i32 %x) {
|
|
; CHECK-LABEL: cttz_i32:
|
|
; CHECK: # BB#0:
|
|
; CHECK-NEXT: bsfl %edi, %eax
|
|
; CHECK-NEXT: retq
|
|
%tmp = call i32 @llvm.cttz.i32( i32 %x, i1 true )
|
|
ret i32 %tmp
|
|
}
|
|
|
|
define i64 @cttz_i64(i64 %x) {
|
|
; CHECK-LABEL: cttz_i64:
|
|
; CHECK: # BB#0:
|
|
; CHECK-NEXT: bsfq %rdi, %rax
|
|
; CHECK-NEXT: retq
|
|
%tmp = call i64 @llvm.cttz.i64( i64 %x, i1 true )
|
|
ret i64 %tmp
|
|
}
|
|
|
|
define i8 @ctlz_i8(i8 %x) {
|
|
; CHECK-LABEL: ctlz_i8:
|
|
; CHECK: # BB#0:
|
|
; CHECK-NEXT: movzbl %dil, %eax
|
|
; CHECK-NEXT: bsrl %eax, %eax
|
|
; CHECK-NEXT: xorl $7, %eax
|
|
; CHECK-NEXT: retq
|
|
%tmp2 = call i8 @llvm.ctlz.i8( i8 %x, i1 true )
|
|
ret i8 %tmp2
|
|
}
|
|
|
|
define i16 @ctlz_i16(i16 %x) {
|
|
; CHECK-LABEL: ctlz_i16:
|
|
; CHECK: # BB#0:
|
|
; CHECK-NEXT: bsrw %di, %ax
|
|
; CHECK-NEXT: xorl $15, %eax
|
|
; CHECK-NEXT: retq
|
|
%tmp2 = call i16 @llvm.ctlz.i16( i16 %x, i1 true )
|
|
ret i16 %tmp2
|
|
}
|
|
|
|
define i32 @ctlz_i32(i32 %x) {
|
|
; CHECK-LABEL: ctlz_i32:
|
|
; CHECK: # BB#0:
|
|
; CHECK-NEXT: bsrl %edi, %eax
|
|
; CHECK-NEXT: xorl $31, %eax
|
|
; CHECK-NEXT: retq
|
|
%tmp = call i32 @llvm.ctlz.i32( i32 %x, i1 true )
|
|
ret i32 %tmp
|
|
}
|
|
|
|
define i64 @ctlz_i64(i64 %x) {
|
|
; CHECK-LABEL: ctlz_i64:
|
|
; CHECK: # BB#0:
|
|
; CHECK-NEXT: bsrq %rdi, %rax
|
|
; CHECK-NEXT: xorq $63, %rax
|
|
; CHECK-NEXT: retq
|
|
%tmp = call i64 @llvm.ctlz.i64( i64 %x, i1 true )
|
|
ret i64 %tmp
|
|
}
|
|
|
|
define i32 @ctlz_i32_cmov(i32 %n) {
|
|
; CHECK-LABEL: ctlz_i32_cmov:
|
|
; CHECK: # BB#0:
|
|
; CHECK-NEXT: bsrl %edi, %ecx
|
|
; CHECK-NEXT: movl $63, %eax
|
|
; CHECK-NEXT: cmovnel %ecx, %eax
|
|
; CHECK-NEXT: xorl $31, %eax
|
|
; CHECK-NEXT: retq
|
|
; Generate a cmov to handle zero inputs when necessary.
|
|
%tmp1 = call i32 @llvm.ctlz.i32(i32 %n, i1 false)
|
|
ret i32 %tmp1
|
|
}
|
|
|
|
define i32 @ctlz_i32_fold_cmov(i32 %n) {
|
|
; CHECK-LABEL: ctlz_i32_fold_cmov:
|
|
; CHECK: # BB#0:
|
|
; CHECK-NEXT: orl $1, %edi
|
|
; CHECK-NEXT: bsrl %edi, %eax
|
|
; CHECK-NEXT: xorl $31, %eax
|
|
; CHECK-NEXT: retq
|
|
; Don't generate the cmovne when the source is known non-zero (and bsr would
|
|
; not set ZF).
|
|
; rdar://9490949
|
|
%or = or i32 %n, 1
|
|
%tmp1 = call i32 @llvm.ctlz.i32(i32 %or, i1 false)
|
|
ret i32 %tmp1
|
|
}
|
|
|
|
define i32 @ctlz_bsr(i32 %n) {
|
|
; CHECK-LABEL: ctlz_bsr:
|
|
; CHECK: # BB#0:
|
|
; CHECK-NEXT: bsrl %edi, %eax
|
|
; CHECK-NEXT: retq
|
|
; Don't generate any xors when a 'ctlz' intrinsic is actually used to compute
|
|
; the most significant bit, which is what 'bsr' does natively.
|
|
%ctlz = call i32 @llvm.ctlz.i32(i32 %n, i1 true)
|
|
%bsr = xor i32 %ctlz, 31
|
|
ret i32 %bsr
|
|
}
|
|
|
|
define i32 @ctlz_bsr_cmov(i32 %n) {
|
|
; CHECK-LABEL: ctlz_bsr_cmov:
|
|
; CHECK: # BB#0:
|
|
; CHECK-NEXT: bsrl %edi, %ecx
|
|
; CHECK-NEXT: movl $63, %eax
|
|
; CHECK-NEXT: cmovnel %ecx, %eax
|
|
; CHECK-NEXT: retq
|
|
; Same as ctlz_bsr, but ensure this happens even when there is a potential
|
|
; zero.
|
|
%ctlz = call i32 @llvm.ctlz.i32(i32 %n, i1 false)
|
|
%bsr = xor i32 %ctlz, 31
|
|
ret i32 %bsr
|
|
}
|