commit 8e5d01cd6efa99a26c8584711a7e8abbcf14c333
author Che-Liang Chiou <clchiou@gmail.com> Thu Feb 10 12:01:24 2011 +0000
committer Che-Liang Chiou <clchiou@gmail.com> Thu Feb 10 12:01:24 2011 +0000
tree 3c066a92a71c255efba8fff8a52c0cb8daa12e4a
parent cbf023d7ececf92071a7175ccf9b42bb15adfe8d

ptx: add passing parameter to kernel functions

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@125279 91177308-0d34-0410-b5e6-96231b3b80d8

lib/Target/PTX/PTXMFInfoExtract.cpp

diff --git a/lib/Target/PTX/PTXMFInfoExtract.cpp b/lib/Target/PTX/PTXMFInfoExtract.cpp
index 68b641b..b37c740 100644
--- a/lib/Target/PTX/PTXMFInfoExtract.cpp
+++ b/lib/Target/PTX/PTXMFInfoExtract.cpp
@@ -67,7 +67,9 @@
 
   // FIXME: This is a slow linear scanning
   for (unsigned reg = PTX::NoRegister + 1; reg < PTX::NUM_TARGET_REGS; ++reg)
-    if (MRI.isPhysRegUsed(reg) && reg != retreg && !MFI->isArgReg(reg))
+    if (MRI.isPhysRegUsed(reg) &&
+        reg != retreg &&
+        (MFI->isKernel() || !MFI->isArgReg(reg)))
       MFI->addLocalVarReg(reg);
 
   // Notify MachineFunctionInfo that I've done adding local var reg