From ae233204174a9d3c41060087ac4e54af10a4b174 Mon Sep 17 00:00:00 2001
From: Evghenii <egaburov@dds.nl>
Date: Mon, 28 Oct 2013 12:10:40 +0100
Subject: [PATCH] added metadata for tasks with nvptx64 target. not tasks are
 kernel callable from host

---
 func.cpp | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/func.cpp b/func.cpp
index b0bbc70f..609e2554 100644
--- a/func.cpp
+++ b/func.cpp
@@ -47,6 +47,7 @@
 #include <stdio.h>
 
 #if defined(LLVM_3_1) || defined(LLVM_3_2)
+  #include <llvm/IR/Metadata.h>
   #include <llvm/LLVMContext.h>
   #include <llvm/Module.h>
   #include <llvm/Type.h>
@@ -54,6 +55,7 @@
   #include <llvm/Intrinsics.h>
   #include <llvm/DerivedTypes.h>
 #else
+  #include <llvm/IR/Metadata.h>
   #include <llvm/IR/LLVMContext.h>
   #include <llvm/IR/Module.h>
   #include <llvm/IR/Type.h>
@@ -316,6 +318,16 @@ Function::emitCode(FunctionEmitContext *ctx, llvm::Function *function,
           taskCountSym2->storagePtr = ctx->AllocaInst(LLVMTypes::Int32Type, "taskCount2");
           ctx->StoreInst(taskCount2, taskCountSym2->storagePtr);
         }
+        else
+        {
+          llvm::NamedMDNode* annotations =
+            m->module->getOrInsertNamedMetadata("nvvm.annotations");
+          llvm::SmallVector<llvm::Value*, 3> av;
+          av.push_back(function);
+          av.push_back(llvm::MDString::get(*g->ctx, "kernel"));
+          av.push_back(llvm::ConstantInt::get(llvm::IntegerType::get(*g->ctx,32), 1));
+          annotations->addOperand(llvm::MDNode::get(*g->ctx, av)); 
+        }
     }
     else {
         // Regular, non-task function