[SYCLomatic]Add a check if it is an CUDA API before reporting warning message (#834)

Signed-off-by: Ni, Wenhui <[email protected]>

Author: wenhuiNi

clang/lib/DPCT/ASTTraversal.cpp

@@ -12305,6 +12305,14 @@ void RecognizeAPINameRule::processFuncCall(const CallExpr *CE,
   std::string Namespace;
   const NamedDecl *ND = dyn_cast<NamedDecl>(CE->getCalleeDecl());
   if (ND) {
+    if (!dpct::DpctGlobalInfo::isInCudaPath(ND->getLocation()) &&
+        !isChildOrSamePath(
+          DpctInstallPath,
+          dpct::DpctGlobalInfo::getLocInfo(ND).first)) {
+      if (!ND->getName().startswith("cudnn") &&
+          !ND->getName().startswith("nccl"))
+        return;
+    }
     const auto *NSD = dyn_cast<NamespaceDecl>(ND->getDeclContext());
     if (NSD && !NSD->isInlineNamespace()) {
       if (dyn_cast<NamespaceDecl>(NSD->getDeclContext())) {

clang/test/dpct/check_std_assert.cu

@@ -0,0 +1,18 @@
+// RUN: dpct --format-range=none -out-root %T/check_std_assert %s --cuda-include-path="%cuda-path/include"
+// RUN: FileCheck %s --match-full-lines --input-file %T/check_std_assert/check_std_assert.dp.cpp
+
+#include <cassert>
+#include <thrust/copy.h>
+
+int main(){
+  // CHECK: int a=1;
+  // CHECK-NEXT: // The function '__assert_fail' with the same name as cuda runtime will be used in assert func
+  // CHECK-NEXT: // Non-CUDA APIs should not be touched or emit warning msgs
+  // CHECK-NEXT: assert(a==0);
+  // CHECK-NEXT: return 1;
+  int a=1;
+  // The function '__assert_fail' with the same name as cuda runtime will be used in assert func
+  // Non-CUDA APIs should not be touched or emit warning msgs
+  assert(a==0);
+  return 1;
+}