diff --git a/llvm/lib/Transforms/DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp b/llvm/lib/Transforms/DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp
index 867c3369c844d3c9e636fefd38ccfd1a845952e6..cb6a2967bbce1117cfda470179bc7cf3ba5d6fce 100644
--- a/llvm/lib/Transforms/DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp
+++ b/llvm/lib/Transforms/DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp
@@ -90,7 +90,7 @@ namespace {
 
     //Functions
     std::string getKernelsModuleName(Module &M);
-    void changeAddrspace(int i);
+    void changeArgAddrspace(Function* F, unsigned i);
     void addCLMetadata(Function* F);
     void writeKernelsModule();
     void transformFunctionToVoid(Function* F);
@@ -841,6 +841,8 @@ namespace {
          re = IItoRemove.rend(); ri != re; ++ri)
       (*ri)->eraseFromParent();
 
+    changeArgAddrspace(F_nvptx, 1);
+    addCLMetadata(F_nvptx);
     DEBUG(errs() << KernelM);
 
     return;
@@ -872,7 +874,7 @@ namespace {
     return mid.append("_kernels.ll");
   }
 
-  void CodeGenTraversal::changeAddrspace(int i) {
+  void CodeGenTraversal::changeArgAddrspace(Function* F, unsigned i) {
   }
 
   void CodeGenTraversal::addCLMetadata(Function* F) {