Migration support for CUDA driver error APIs and cudaErrorr_enum

Signed-off-by: Teja Alaghari <[email protected]>
oneapi-src · Dec 19, 2023 · c1a018b · c1a018b
1 parent b6849e5
commit c1a018b
Show file tree

Hide file tree

Showing 7 changed files with 42 additions and 6 deletions.
diff --git a/clang/lib/DPCT/APINames.inc b/clang/lib/DPCT/APINames.inc
@@ -84,6 +84,10 @@ ENTRY(cudaGetErrorString, cudaGetErrorString, true, NO_FLAG, P0, "Successful")
 ENTRY(cudaGetLastError, cudaGetLastError, true, NO_FLAG, P0, "DPCT1010")
 ENTRY(cudaPeekAtLastError, cudaPeekAtLastError, true, NO_FLAG, P0, "DPCT1026/DPCT1027")
 
+// error handling functions of driver API
+ENTRY(cuGetErrorName, cuGetErrorName, true, NO_FLAG, P0, "Successful")
+ENTRY(cuGetErrorString, cuGetErrorString, true, NO_FLAG, P0, "Successful")
+
 // stream management functions of API
 ENTRY(cudaCtxResetPersistingL2Cache, cudaCtxResetPersistingL2Cache, true, NO_FLAG, P7, "DPCT1026/DPCT1027")
 ENTRY(cudaStreamAddCallback, cudaStreamAddCallback, true, NO_FLAG, P0, "Successful")

diff --git a/clang/lib/DPCT/APINamesErrorHandling.inc b/clang/lib/DPCT/APINamesErrorHandling.inc
@@ -22,6 +22,14 @@ WARNING_FACTORY_ENTRY(
                           "\"cudaGetErrorString is not supported\"/*", "*/"),
     Diagnostics::TRNA_WARNING_ERROR_HANDLING_API_COMMENTED)
 
+WARNING_FACTORY_ENTRY(
+    "cuGetErrorString",
+    INSERT_AROUND_FACTORY(CALL_FACTORY_ENTRY("cuGetErrorString",
+                                             CALL("cuGetErrorString",
+                                                  ARG_WC(0), ARG_WC(1))),
+                          "\"cuGetErrorString is not supported\"/*", "*/"),
+    Diagnostics::TRNA_WARNING_ERROR_HANDLING_API_COMMENTED)
+
 WARNING_FACTORY_ENTRY(
     "cudaGetErrorName",
     INSERT_AROUND_FACTORY(CALL_FACTORY_ENTRY("cudaGetErrorName",
@@ -30,6 +38,14 @@ WARNING_FACTORY_ENTRY(
                           "\"cudaGetErrorName is not supported\"/*", "*/"),
     Diagnostics::TRNA_WARNING_ERROR_HANDLING_API_COMMENTED)
 
+WARNING_FACTORY_ENTRY(
+    "cuGetErrorName",
+    INSERT_AROUND_FACTORY(CALL_FACTORY_ENTRY("cuGetErrorName",
+                                             CALL("cuGetErrorName",
+                                                  ARG_WC(0), ARG_WC(1))),
+                          "\"cuGetErrorName is not supported\"/*", "*/"),
+    Diagnostics::TRNA_WARNING_ERROR_HANDLING_API_COMMENTED)
+
 CONDITIONAL_FACTORY_ENTRY(
     checkIsCallExprOnly(),
     WARNING_FACTORY_ENTRY(

diff --git a/clang/lib/DPCT/ASTTraversal.cpp b/clang/lib/DPCT/ASTTraversal.cpp
@@ -3476,6 +3476,11 @@ void ErrorConstantsRule::runRule(const MatchFinder::MatchResult &Result) {
     }
   }
 
+  auto Search = EnumConstantRule::EnumNamesMap.find(EC->getDeclName().getAsString());
+  if (Search != EnumConstantRule::EnumNamesMap.end()) {
+    Repl = Search->second->NewName;
+  }
+
   emplaceTransformation(new ReplaceStmt(DE, Repl));
 }
 
@@ -6251,7 +6256,8 @@ void FunctionCallRule::registerMatcher(MatchFinder &MF) {
         "cudaDeviceGetAttribute", "cudaDeviceGetP2PAttribute",
         "cudaDeviceGetPCIBusId", "cudaGetDevice", "cudaDeviceSetLimit",
         "cudaGetLastError", "cudaPeekAtLastError", "cudaDeviceSynchronize",
-        "cudaThreadSynchronize", "cudnnGetErrorString", "cudaGetErrorString", "cudaGetErrorName",
+        "cudaThreadSynchronize", "cudnnGetErrorString", "cudaGetErrorString",
+        "cuGetErrorString", "cudaGetErrorName", "cuGetErrorName",
         "cudaDeviceSetCacheConfig", "cudaDeviceGetCacheConfig", "clock",
         "cudaOccupancyMaxPotentialBlockSize", "cudaThreadSetLimit",
         "cudaFuncSetCacheConfig", "cudaThreadExit", "cudaDeviceGetLimit",
@@ -6562,7 +6568,9 @@ void FunctionCallRule::runRule(const MatchFinder::MatchResult &Result) {
   } else if (FuncName == "cudaGetLastError" ||
              FuncName == "cudaPeekAtLastError" ||
              FuncName == "cudaGetErrorString" ||
-             FuncName == "cudaGetErrorName") {
+             FuncName == "cuGetErrorString" ||
+             FuncName == "cudaGetErrorName" ||
+             FuncName == "cuGetErrorName") {
     ExprAnalysis EA(CE);
     emplaceTransformation(EA.getReplacement());
     EA.applyAllSubExprRepl();

diff --git a/clang/lib/DPCT/MapNames.cpp b/clang/lib/DPCT/MapNames.cpp
@@ -101,6 +101,9 @@ void MapNames::setExplicitNamespaceMap() {
       {"cudaError",
        std::make_shared<TypeNameRule>(getDpctNamespace() + "err0",
                                       HelperFeatureEnum::device_ext)},
+      {"cudaError_enum",
+       std::make_shared<TypeNameRule>(getDpctNamespace() + "error_code",
+                                      HelperFeatureEnum::device_ext)},
       {"CUjit_option", std::make_shared<TypeNameRule>("int")},
       {"CUresult", std::make_shared<TypeNameRule>("int")},
       {"CUcontext", std::make_shared<TypeNameRule>("int")},
@@ -891,6 +894,11 @@ void MapNames::setExplicitNamespaceMap() {
       {"cudaDevAttrConcurrentManagedAccess",
        std::make_shared<EnumNameRule>(
            "get_info<sycl::info::device::usm_shared_allocations>")},
+      // enum cudaError_enum Type
+      {"cudaErrorInvalidConfiguration",
+       std::make_shared<EnumNameRule>(
+           getDpctNamespace() + "default_error",
+           HelperFeatureEnum::device_ext)},
       // enum Memcpy Kind
       {"cudaMemcpyHostToHost",
        std::make_shared<EnumNameRule>(getDpctNamespace() + "host_to_host")},

diff --git a/clang/test/dpct/cpp_test.cpp b/clang/test/dpct/cpp_test.cpp
@@ -188,7 +188,7 @@ void test_simple_ifs() {
 // CHECK-NEXT:            return "cudaErrorLaunchOutOfResources";
 // CHECK-NEXT:        case {{[0-9]+}}:
 // CHECK-NEXT:            return "cudaErrorInvalidDeviceFunction";
-// CHECK-NEXT:        case 9:
+// CHECK-NEXT:        case dpct::default_error:
 // CHECK-NEXT:            return "cudaErrorInvalidConfiguration";
 // CHECK-NEXT:        case {{[0-9]+}}:
 // CHECK-NEXT:            return "cudaErrorInvalidDevice";

diff --git a/clang/test/dpct/error-handling-constants.cu b/clang/test/dpct/error-handling-constants.cu
@@ -27,7 +27,7 @@
 // CHECK-NEXT:            return "cudaErrorLaunchOutOfResources";
 // CHECK-NEXT:        case {{[0-9]+}}:
 // CHECK-NEXT:            return "cudaErrorInvalidDeviceFunction";
-// CHECK-NEXT:        case 9:
+// CHECK-NEXT:        case dpct::default_error:
 // CHECK-NEXT:            return "cudaErrorInvalidConfiguration";
 // CHECK-NEXT:        case {{[0-9]+}}:
 // CHECK-NEXT:            return "cudaErrorInvalidDevice";

diff --git a/clang/test/dpct/types001.cu b/clang/test/dpct/types001.cu
@@ -318,8 +318,8 @@ int main(int argc, char **argv) {
   a = sizeof(fftresult);
   a = sizeof fftresult;
 
-  //CHECK:cudaError_enum error_enum;
-  //CHECK-NEXT:a = sizeof(cudaError_enum);
+  //CHECK:dpct::error_code error_enum;
+  //CHECK-NEXT:a = sizeof(dpct::error_code);
   //CHECK-NEXT:a = sizeof(error_enum);
   //CHECK-NEXT:a = sizeof error_enum;
   cudaError_enum error_enum;