[SYCLomatic] Add migration support for cuStreamGetCtx (#2624)

Author: Deepak Raj H R

clang/lib/DPCT/RulesLang/APINamesStream.inc

@@ -31,6 +31,12 @@ ASSIGNABLE_FACTORY(HEADER_INSERT_FACTORY(
                                    MEMBER_CALL(ARG(0), true, "wait"),
                                    CALL(ARG(1), ARG(0), ARG("0"), ARG(2)))))))
 
+ASSIGNABLE_FACTORY(FEATURE_REQUEST_FACTORY(
+    HelperFeatureEnum::device_ext,
+    ASSIGN_FACTORY_ENTRY("cuStreamGetCtx", DEREF(1),
+                         CALL(MapNames::getDpctNamespace() + "get_device_id",
+                              MEMBER_CALL(ARG(0), true, "get_device")))))
+
 ASSIGNABLE_FACTORY(CONDITIONAL_FACTORY_ENTRY(
     UseExtGraph,
     CALL_FACTORY_ENTRY("cudaStreamBeginCapture",

clang/lib/DPCT/RulesLang/RulesLang.cpp

@@ -4178,7 +4178,7 @@ void StreamAPICallRule::registerMatcher(MatchFinder &MF) {
         "cudaStreamIsCapturing", "cudaStreamQuery", "cudaStreamWaitEvent",
         "cudaStreamAddCallback", "cuStreamCreate", "cuStreamSynchronize",
         "cuStreamWaitEvent", "cuStreamDestroy_v2", "cuStreamAttachMemAsync",
-        "cuStreamAddCallback", "cuStreamQuery");
+        "cuStreamAddCallback", "cuStreamQuery", "cuStreamGetCtx");
   };
 
   MF.addMatcher(

clang/lib/DPCT/SrcAPI/APINames.inc

@@ -1783,7 +1783,7 @@ ENTRY(cuStreamEndCapture, cuStreamEndCapture, false, NO_FLAG, P4, "comment")
 ENTRY(cuStreamGetAttribute, cuStreamGetAttribute, false, NO_FLAG, P7, "comment")
 ENTRY(cuStreamGetCaptureInfo, cuStreamGetCaptureInfo_v2, false, NO_FLAG, P7, "comment")
 ENTRY(cuStreamGetCaptureInfo_v3, cuStreamGetCaptureInfo_v3, false, NO_FLAG, P7, "comment")
-ENTRY(cuStreamGetCtx, cuStreamGetCtx, false, NO_FLAG, P4, "comment")
+ENTRY(cuStreamGetCtx, cuStreamGetCtx, true, NO_FLAG, P4, "Successful")
 ENTRY(cuStreamGetCtx_v2, cuStreamGetCtx_v2, false, NO_FLAG, P4, "comment")
 ENTRY(cuStreamGetFlags, cuStreamGetFlags, false, NO_FLAG, P4, "comment")
 ENTRY(cuStreamGetId, cuStreamGetId, false, NO_FLAG, P4, "comment")

clang/test/dpct/driver-stream-and-event.cu

@@ -183,3 +183,9 @@ unsigned getEventFlags(bool enabledSyncBlock) {
 
   return flags;
 }
+
+// CHECK: void test_stream_and_context(dpct::queue_ptr stream, int &context) {
+void test_stream_and_context(CUstream stream, CUcontext& context) {
+  // CHECK: context = dpct::get_device_id(stream->get_device());
+  cuStreamGetCtx(stream, &context);
+}