[SYCLomatic] Add query-api-mapping for 1 runtime API cudaEventRecordWithFlags

Signed-off-by: chenwei.sun <chenwei.sun@intel.com>

Author: tomflinda

clang/examples/DPCT/Runtime/cudaEventRecordWithFlags.cu

@@ -0,0 +1,6 @@
+void test(cudaEvent_t event, cudaStream_t stream) {
+  // Start
+  cudaEventRecordWithFlags(event /*cudaEvent_t*/, stream /*cudaStream_t*/,
+                           cudaEventRecordDefault /*unsigned int*/);
+  // End
+}

clang/lib/DPCT/RulesLang/RulesLang.cpp

@@ -3276,6 +3276,19 @@ void EventAPICallRule::handleEventRecordWithProfilingDisabled(
   static std::set<std::pair<const Decl *, std::string>> DeclDupFilter;
   auto &SM = DpctGlobalInfo::getSourceManager();
 
+  int NumArgs = CE->getNumArgs();
+  if (NumArgs == 3) { // Special process for cudaEventRecordWithFlags().
+    auto APIName = CE->getDirectCallee()->getNameInfo().getName().getAsString();
+    const Expr *SecArg = CE->getArg(2);
+    ExprAnalysis Arg2EA(SecArg);
+    auto Arg2Name = Arg2EA.getReplacedString();
+    if (Arg2Name != "cudaEventRecordDefault") {
+      report(CE->getBeginLoc(), Diagnostics::NOT_SUPPORTED_PARAMETER, false,
+             APIName, "parameter " + Arg2Name + " is unsupported");
+      return;
+    }
+  }
+
   const ValueDecl *MD = nullptr;
   if ((MD = getDecl(CE->getArg(0))) == nullptr)
     return;

clang/test/dpct/query_api_mapping/Runtime/test.cu

@@ -519,3 +519,11 @@
 // CUDARUNTIMEGETVERSION-NEXT:   cudaRuntimeGetVersion(pi /*int **/);
 // CUDARUNTIMEGETVERSION-NEXT: Is migrated to:
 // CUDARUNTIMEGETVERSION-NEXT:   *pi = dpct::get_major_version(dpct::get_current_device());
+
+// RUN: dpct --cuda-include-path="%cuda-path/include" --query-api-mapping=cudaEventRecordWithFlags | FileCheck %s -check-prefix=cudaEventRecordWithFlags
+// cudaEventRecordWithFlags: CUDA API:
+// cudaEventRecordWithFlags-NEXT:   cudaEventRecordWithFlags(event /*cudaEvent_t*/, stream /*cudaStream_t*/,
+// cudaEventRecordWithFlags-NEXT:                            cudaEventRecordDefault /*unsigned int*/);
+// cudaEventRecordWithFlags-NEXT: Is migrated to:
+// cudaEventRecordWithFlags-NEXT:     *event = dpct::get_in_order_queue().ext_oneapi_submit_barrier();
+

clang/test/dpct/query_api_mapping/test_all.cu

@@ -1446,6 +1446,7 @@
 // CHECK-NEXT: cudaEventElapsedTime
 // CHECK-NEXT: cudaEventQuery
 // CHECK-NEXT: cudaEventRecord
+// CHECK-NEXT: cudaEventRecordWithFlags
 // CHECK-NEXT: cudaEventSynchronize
 // CHECK-NEXT: cudaExternalMemoryGetMappedBuffer
 // CHECK-NEXT: cudaExternalMemoryGetMappedMipmappedArray