[llvm-commits] [poolalloc] r155767 - /poolalloc/trunk/lib/AssistDS/DSNodeEquivs.cpp

[Matthew Wala]

Author: wala1
Date: Sat Apr 28 15:33:02 2012
New Revision: 155767

URL: http://llvm.org/viewvc/llvm-project?rev=155767&view=rev
Log:
Make DSNodeEquivs look up arguments.

Modified:
    poolalloc/trunk/lib/AssistDS/DSNodeEquivs.cpp

Modified: poolalloc/trunk/lib/AssistDS/DSNodeEquivs.cpp
URL: http://llvm.org/viewvc/llvm-project/poolalloc/trunk/lib/AssistDS/DSNodeEquivs.cpp?rev=155767&r1=155766&r2=155767&view=diff
==============================================================================
--- poolalloc/trunk/lib/AssistDS/DSNodeEquivs.cpp (original)
+++ poolalloc/trunk/lib/AssistDS/DSNodeEquivs.cpp Sat Apr 28 15:33:02 2012
@@ -225,6 +225,9 @@
   } else if (isa<Instruction>(V)) {
     const Function *Parent = cast<Instruction>(V)->getParent()->getParent();
     NHForV = &TDDS.getDSGraph(*Parent)->getNodeForValue(V);
+  } else if (isa<Argument>(V)) {
+    const Function *Parent = cast<Argument>(V)->getParent();
+    NHForV = &TDDS.getDSGraph(*Parent)->getNodeForValue(V);
   }
 
   if (NHForV == 0 || NHForV->isNull())