From 873f25a5887ac930f7589e3a8e89cec59901b580 Mon Sep 17 00:00:00 2001
From: aejjeh <aejjeh@illinois.edu>
Date: Wed, 18 Dec 2019 18:08:10 -0600
Subject: [PATCH] Update DFG2LLVM_NVPTX.cpp to fix callinst of a function
 within the kernel.

---
 llvm/lib/Transforms/DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/llvm/lib/Transforms/DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp b/llvm/lib/Transforms/DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp
index 3670c2ef55..d8be759932 100644
--- a/llvm/lib/Transforms/DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp
+++ b/llvm/lib/Transforms/DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp
@@ -1465,6 +1465,10 @@ void CGT_NVPTX::codeGen(DFLeafNode* N) {
         Function* newCalleeF = CloneFunction(calleeF, VMap);
         newCalleeF->removeFromParent(); //TODO: MARIA check
         KernelM->getFunctionList().push_back(newCalleeF);
+        CallInst *CInew = CallInst::Create(newCalleeF, CI->getArgOperand(0), CI->getName(), CI);
+        CI->replaceAllUsesWith(CInew);
+        CItoRemove.push_back(CI);
+
       }
       //TODO: how to handle address space qualifiers in load/store
     }
-- 
GitLab