This patch tackles PR46235 to improve the code generated for bit tests
on x86_64 by making more use of the bt instruction.  Currently, GCC emits
bt instructions when followed by condition jumps (thanks to Uros'
splitters).
This patch adds splitters in i386.md, to catch the cases where bt is
followed
by a conditional move (as in the original report), or by a setc/setnc (as in
comment 5 of the Bugzilla PR).

With this patch, the motivating function in the original PR

int foo(int a, int x, int y) {
    if (a & (1 << x))
       return a;
   return 1;
}

which with -O2 on mainline generates:

foo:    movl    %edi, %eax
        movl    %esi, %ecx
        sarl    %cl, %eax
        testb   $1, %al
        movl    $1, %eax
        cmovne  %edi, %eax
        ret

now generates:
foo:    btl     %esi, %edi
        movl    $1, %eax
        cmovc   %edi, %eax
        ret

Likewise, IsBitSet1 (from comment 5)

bool IsBitSet1(unsigned char byte, int index) {
    return (byte & (1<<index)) != 0;
}

Before:
        movzbl  %dil, %eax
        movl    %esi, %ecx
        sarl    %cl, %eax
        andl    $1, %eax
        ret

After:
        movzbl  %dil, %edi
        btl     %esi, %edi
        setc    %al
        ret

[Identical code is generated for comment 5's IsBitSet2]
bool IsBitSet2(unsigned char byte, int index) {
    return (byte >> index) & 1;
}

And finally to demonstrate the corner cases also handled,

int IsBitClr(long long dword, int index) {
    return (dword & (1LL<<index)) == 0;
}

Before:
        movq    %rdi, %rax
        movl    %esi, %ecx
        sarq    %cl, %rax
        notq    %rax
        andl    $1, %eax
        ret

After:
        xorl    %eax, %eax
        btq     %rsi, %rdi
        setnc   %al
        ret

According to Agner Fog, SAR/SHR r,cl takes 2 cycles on skylake,
where BT r,r takes only one, so the performance improvements on
recent hardware may be more significant than implied by just the
reduced number of instructions.  I've avoided transforming cases
(such as btsi_setcsi) where using bt sequences may not be a clear
win (over sarq/andl).

This patch has been tested on x86_64-pc-linux-gnu with a "make
bootstrap" and "make -k check" with no new failures.

Ok for mainline?

2010-06-15  Roger Sayle  <ro...@nextmovesoftware.com>

gcc/ChangeLog
        PR rtl-optimization/46235
        * config/i386/i386.md: New define_split for bt followed by cmov.
        (*bt<mode>_setcqi): New define_insn_and_split for bt followed by
setc.
        (*bt<mode>_setncqi): New define_insn_and_split for bt then setnc.
        (*bt<mode>_setnc<mode>): New define_insn_and_split for bt followed
        by setnc with zero extension.

gcc/testsuite/ChangeLog
        PR rtl-optimization/46235
        * gcc.target/i386/bt-5.c: New test.
        * gcc.target/i386/bt-6.c: New test.
        * gcc.target/i386/bt-7.c: New test.

Roger
--
Roger Sayle
NextMove Software 
Cambridge, UK

diff --git a/gcc/config/i386/i386.md b/gcc/config/i386/i386.md
index 7743c61e..1ace33a 100644
--- a/gcc/config/i386/i386.md
+++ b/gcc/config/i386/i386.md
@@ -12794,6 +12794,100 @@
   operands[0] = shallow_copy_rtx (operands[0]);
   PUT_CODE (operands[0], reverse_condition (GET_CODE (operands[0])));
 })
+
+;; Help combine recognize bt followed by cmov
+(define_split
+  [(set (match_operand:SWI248 0 "register_operand")
+       (if_then_else:SWI248
+        (ne
+         (zero_extract:SWI48
+          (match_operand:SWI48 1 "register_operand")
+          (const_int 1)
+          (zero_extend:SI (match_operand:QI 2 "register_operand")))
+         (const_int 0))
+        (match_operand:SWI248 3 "nonimmediate_operand")
+        (match_operand:SWI248 4 "nonimmediate_operand")))]
+  "TARGET_USE_BT && TARGET_CMOVE
+   && !(MEM_P (operands[3]) && MEM_P (operands[4]))
+   && ix86_pre_reload_split ()"
+  [(set (reg:CCC FLAGS_REG)
+       (compare:CCC
+        (zero_extract:SWI48 (match_dup 1) (const_int 1) (match_dup 2))
+        (const_int 0)))
+   (set (match_dup 0)
+       (if_then_else:SWI248 (eq (reg:CCC FLAGS_REG) (const_int 0))
+                            (match_dup 3)
+                            (match_dup 4)))]
+{
+  operands[2] = lowpart_subreg (SImode, operands[2], QImode);
+})
+
+;; Help combine recognize bt followed by setc
+(define_insn_and_split "*bt<mode>_setcqi"
+  [(set (subreg:SWI48 (match_operand:QI 0 "register_operand") 0)
+        (zero_extract:SWI48
+         (match_operand:SWI48 1 "register_operand")
+         (const_int 1)
+         (zero_extend:SI (match_operand:QI 2 "register_operand"))))
+   (clobber (reg:CC FLAGS_REG))]
+  "TARGET_USE_BT && ix86_pre_reload_split ()"
+  "#"
+  "&& 1"
+  [(set (reg:CCC FLAGS_REG)
+        (compare:CCC
+         (zero_extract:SWI48 (match_dup 1) (const_int 1) (match_dup 2))
+         (const_int 0)))
+   (set (match_dup 0)
+        (eq:QI (reg:CCC FLAGS_REG) (const_int 0)))]
+{
+  operands[2] = lowpart_subreg (SImode, operands[2], QImode);
+})
+
+;; Help combine recognize bt followed by setnc
+(define_insn_and_split "*bt<mode>_setncqi"
+  [(set (match_operand:QI 0 "register_operand")
+       (and:QI
+        (not:QI
+         (subreg:QI
+          (lshiftrt:SWI48 (match_operand:SWI48 1 "register_operand")
+                          (match_operand:QI 2 "register_operand")) 0))
+        (const_int 1)))
+   (clobber (reg:CC FLAGS_REG))]
+  "TARGET_USE_BT && ix86_pre_reload_split ()"
+  "#"
+  "&& 1"
+  [(set (reg:CCC FLAGS_REG)
+        (compare:CCC
+         (zero_extract:SWI48 (match_dup 1) (const_int 1) (match_dup 2))
+         (const_int 0)))
+   (set (match_dup 0)
+        (ne:QI (reg:CCC FLAGS_REG) (const_int 0)))]
+{
+  operands[2] = lowpart_subreg (SImode, operands[2], QImode);
+})
+
+(define_insn_and_split "*bt<mode>_setnc<mode>"
+  [(set (match_operand:SWI48 0 "register_operand")
+       (and:SWI48
+        (not:SWI48
+         (lshiftrt:SWI48 (match_operand:SWI48 1 "register_operand")
+                         (match_operand:QI 2 "register_operand")))
+        (const_int 1)))
+   (clobber (reg:CC FLAGS_REG))]
+  "TARGET_USE_BT && ix86_pre_reload_split ()"
+  "#"
+  "&& 1"
+  [(set (reg:CCC FLAGS_REG)
+        (compare:CCC
+         (zero_extract:SWI48 (match_dup 1) (const_int 1) (match_dup 2))
+         (const_int 0)))
+   (set (match_dup 3)
+        (ne:QI (reg:CCC FLAGS_REG) (const_int 0)))
+   (set (match_dup 0) (zero_extend:SWI48 (match_dup 3)))]
+{
+  operands[2] = lowpart_subreg (SImode, operands[2], QImode);
+  operands[3] = gen_reg_rtx (QImode);
+})
 
 ;; Store-flag instructions.
 
/* PR rtl-optimization/46235 */
/* { dg-do compile { target lp64 } } */
/* { dg-options "-O2 -mtune=core2" } */

int foo (int a, int x, int y)
{
  if (a & (1<<x))
    return a;
  return 1;
}

int bar_ww (int a, int x, int y, int z)
{
  return (a & (1<<x)) ? y : z;
}

int bar_lw (long long a, int x, int y, int z)
{
  return (a & (1LL<<x)) ? y : z;
}

long long bar_wl (int a, int x, long long y, long long z)
{
  return (a & (1<<x)) ? y : z;
}

long long bar_ll (long long a, int x, long long y, long long z)
{
  return (a & (1LL<<x)) ? y : z;
}

short bar_ws (int a, int x, short y, short z)
{
  return (a & (1<<x)) ? y : z;
}

short bar_ls (long long a, int x, short y, short z)
{
  return (a & (1LL<<x)) ? y : z;
}

/* { dg-final { scan-assembler-times "bt\[lq\]\[ \t\]" 7 } } */
/* { dg-final { scan-assembler-not "sar\[lq\]\[ \t\]" } } */

/* PR rtl-optimization/46235 */
/* { dg-do compile } */
/* { dg-options "-O2 -mtune=core2" } */

unsigned char set1_bb (unsigned char x, int y)
{
  return (x & (1<<y)) != 0;
}

unsigned char set2_bb (unsigned char x, int y)
{
  return (x >> y) & 1;
}

unsigned char set1_wb (int x, int y)
{
  return (x & (1<<y)) != 0;
}

unsigned char set2_wb (int x, int y)
{
  return (x >> y) & 1;
}

unsigned char clr1_bb (unsigned char x, int y)
{
  return (x & (1<<y)) == 0;
}

unsigned char clr2_bb (unsigned char x, int y)
{
  return !((x >> y) & 1);
}

unsigned char clr1_wb (int x, int y)
{
  return (x & (1<<y)) == 0;
}

unsigned char clr2_wb (int x, int y)
{
  return !((x >> y) & 1);
}

int clr1_bw (unsigned char x, int y)
{
  return (x & (1<<y)) == 0;
}

int clr2_bw (unsigned char x, int y)
{
  return !((x >> y) & 1);
}

int clr1_ww (int x, int y)
{
  return (x & (1<<y)) == 0;
}

int clr2_ww (int x, int y)
{
  return !((x >> y) & 1);
}

/* { dg-final { scan-assembler-times "bt\[lq\]\[ \t\]" 12 } } */
/* { dg-final { scan-assembler-not "sar\[lq\]\[ \t\]" } } */
/* { dg-final { scan-assembler-not "and\[lq\]\[ \t\]" } } */
/* { dg-final { scan-assembler-not "not\[lq\]\[ \t\]" } } */

/* PR rtl-optimization/46235 */
/* { dg-do compile { target lp64 } } */
/* { dg-options "-O2 -mtune=core2" } */

unsigned char set1_lb (long long x, int y)
{
  return (x & (1LL<<y)) != 0;
}

unsigned char set2_lb (long long x, int y)
{
  return (x >> y) & 1;
}

unsigned char clr1_lb (long long x, int y)
{
  return (x & (1LL<<y)) == 0;
}

unsigned char clr2_lb (long long x, int y)
{
  return !((x >> y) & 1);
}

int clr1_lw (long long x, int y)
{
  return (x & (1LL<<y)) == 0;
}

int clr2_lw (long long x, int y)
{
  return !((x >> y) & 1);
}

long long clr1_bl (unsigned char x, int y)
{
  return (x & (1<<y)) == 0;
}

long long clr2_bl (unsigned char x, int y)
{
  return !((x >> y) & 1);
}

long long clr1_wl (int x, int y)
{
  return (x & (1<<y)) == 0;
}

long long clr2_wl (int x, int y)
{
  return !((x >> y) & 1);
}

long long clr1_ll (long long x, int y)
{
  return (x & (1LL<<y)) == 0;
}

long long clr2_ll (long long x, int y)
{
  return !((x >> y) & 1);
}

/* { dg-final { scan-assembler-times "bt\[lq\]\[ \t\]" 12 } } */
/* { dg-final { scan-assembler-not "sar\[lq\]\[ \t\]" } } */
/* { dg-final { scan-assembler-not "and\[lq\]\[ \t\]" } } */
/* { dg-final { scan-assembler-not "not\[lq\]\[ \t\]" } } */

Reply via email to