Add warning for cudaStreamCaptureStatusInvalidated enum value

Signed-off-by: Ahmed, Daiyaan <[email protected]>
oneapi-src · May 3, 2024 · 3b99638 · 3b99638
1 parent 4b6d51d
commit 3b99638
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/clang/lib/DPCT/ASTTraversal.cpp b/clang/lib/DPCT/ASTTraversal.cpp
@@ -3389,6 +3389,9 @@ void EnumConstantRule::runRule(const MatchFinder::MatchResult &Result) {
       EnumName == "cudaComputeModeExclusiveProcess") {
     handleComputeMode(EnumName, E);
     return;
+  } else if (EnumName == "cudaStreamCaptureStatusInvalidated") {
+    report(E->getBeginLoc(), Diagnostics::KNOWN_UNSUPPORTED_TYPE, true,
+           "cudaStreamCaptureStatus::cudaStreamCaptureStatusInvalidated");
   } else if (auto ET = dyn_cast<EnumType>(E->getType())) {
     if (auto ETD = ET->getDecl()) {
       auto EnumTypeName = ETD->getName().str();

diff --git a/clang/lib/DPCT/MapNames.cpp b/clang/lib/DPCT/MapNames.cpp
@@ -1049,8 +1049,8 @@ void MapNames::setExplicitNamespaceMap() {
       {"cudaStreamCaptureStatusInvalidated",
        std::make_shared<EnumNameRule>(
            DpctGlobalInfo::useExtGraphs()
-               ? getClNamespace() +
-                     "ext::oneapi::experimental::queue_state::executing"
+               ? "("+ getClNamespace() +
+                     "ext::oneapi::experimental::queue_state) -1"
                : "0")},
       // enum CUmem_advise_enum
       {"CU_MEM_ADVISE_SET_READ_MOSTLY", std::make_shared<EnumNameRule>("0")},

diff --git a/clang/test/dpct/cuda-stream-api-cuda10-after.cu b/clang/test/dpct/cuda-stream-api-cuda10-after.cu
@@ -19,11 +19,16 @@ int main() {
   cudaStreamCaptureStatus captureStatus = cudaStreamCaptureStatusNone;
   captureStatus = cudaStreamCaptureStatusActive;
   captureStatus = cudaStreamCaptureStatusInvalidated;
+
   // CHECK: /*
   // CHECK-NEXT: DPCT1027:{{[0-9]+}}: The call to cudaStreamIsCapturing was replaced with 0 because SYCL currently does not support capture operations on queues.
   // CHECK-NEXT: */
   // CHECK: MY_ERROR_CHECKER(0);
   MY_ERROR_CHECKER(cudaStreamIsCapturing(s0, &captureStatus));
 
+  if (captureStatus == cudaStreamCaptureStatusInvalidated) {
+    return -1;
+  }
+
   return 0;
-}
+}