[llvm-commits] CVS: llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp

[Andrew Lenharth]

Changes in directory llvm/lib/CodeGen/SelectionDAG:

LegalizeDAG.cpp updated: 1.95 -> 1.96
---
Log message:

ctpop lowering in legalize

---
Diffs of the changes:  (+33 -1)

 LegalizeDAG.cpp |   34 +++++++++++++++++++++++++++++++++-
 1 files changed, 33 insertions(+), 1 deletion(-)


Index: llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp
diff -u llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp:1.95 llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp:1.96
--- llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp:1.95	Wed May  4 14:11:05 2005
+++ llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp	Thu May  5 10:55:21 2005
@@ -1027,7 +1027,39 @@
     case TargetLowering::Custom:
       assert(0 && "Cannot custom handle this yet!");
     case TargetLowering::Expand:
-      assert(0 && "Cannot expand this yet!");
+      switch(Node->getOpcode())
+      {
+      case ISD::CTPOP: {
+        static const uint64_t mask[6][9] = {
+          {0, 0x55, 0x5555, 0, 0x55555555, 0, 0, 0, 0x5555555555555555ULL},
+          {0, 0x33, 0x3333, 0, 0x33333333, 0, 0, 0, 0x3333333333333333ULL},
+          {0, 0x0F, 0x0F0F, 0, 0x0F0F0F0F, 0, 0, 0, 0x0F0F0F0F0F0F0F0FULL},
+          {0,    0, 0x00FF, 0, 0x00FF00FF, 0, 0, 0, 0x00FF00FF00FF00FFULL},
+          {0,    0,      0, 0, 0x0000FFFF, 0, 0, 0, 0x0000FFFF0000FFFFULL},
+          {0,    0,      0, 0,          0, 0, 0, 0, 0x00000000FFFFFFFFULL}};
+        MVT::ValueType VT = Tmp1.getValueType();
+        int len = getSizeInBits(VT);
+        for (int i = 0; (1 << i) <= (len / 2); ++i) {
+          //x = (x & mask[i][len/8]) + (x >> (1 << i) & mask[i][len/8])
+          Tmp2 = DAG.getConstant(mask[i][len/8], VT);
+          Tmp3 = DAG.getConstant(1 << i, VT);
+          Tmp1 = DAG.getNode(ISD::ADD, VT, 
+                             DAG.getNode(ISD::AND, VT, Tmp1, Tmp2),
+                             DAG.getNode(ISD::AND, VT,
+                                         DAG.getNode(ISD::SRL, VT, Tmp1, Tmp3),
+                                         Tmp2));
+        }
+        Result = Tmp1;
+        break;
+      }
+//       case ISD::CTTZ:
+//         break;
+//       case ISD::CTLZ:
+//         break;
+      default:
+        assert(0 && "Cannot expand this yet!");
+        break;
+      }
       break;
     }
     break;