add IsCUDAFunction

Author: Vipul-Cariappa

include/CppInterOp/CppInterOp.h

@@ -281,6 +281,9 @@ CPPINTEROP_API bool IsNamespace(TCppScope_t scope);
 /// Checks if the scope is a class or not.
 CPPINTEROP_API bool IsClass(TCppScope_t scope);
 
+/// Checks if the scope is a CUDA function or not.
+CPPINTEROP_API bool IsCUDAFunction(TCppScope_t scope);
+
 /// Checks if the scope is a function.
 CPPINTEROP_API bool IsFunction(TCppScope_t scope);
 

lib/CppInterOp/CppInterOp.cpp

@@ -304,6 +304,17 @@ bool IsNamespace(TCppScope_t scope) {
   return isa<NamespaceDecl>(D);
 }
 
+bool IsCUDAFunction(TCppScope_t scope) {
+  Decl* D = static_cast<Decl*>(scope);
+  if (auto* FD = llvm::dyn_cast<FunctionDecl>(D)) {
+    for (const auto& i : FD->attrs()) {
+      if (i->getKind() == clang::attr::Kind::CUDAGlobal)
+        return true;
+    }
+  }
+  return false;
+}
+
 bool IsClass(TCppScope_t scope) {
   Decl* D = static_cast<Decl*>(scope);
   return isa<CXXRecordDecl>(D);