[SYCLomatic] Enable migration of cudnnGetVersion (#771)

Signed-off-by: Wang, Hao3 <[email protected]>

Author: intwanghao

clang/lib/DPCT/APINamesCUDNN.inc

@@ -7,6 +7,11 @@
 //===-----------------------------------------------------------------===//
 
 // Base API
+FEATURE_REQUEST_FACTORY(HelperFeatureEnum::DnnlUtils_get_version,
+                        CALL_FACTORY_ENTRY("cudnnGetVersion",
+                                           CALL(MapNames::getDpctNamespace() +
+                                                "dnnl::get_version")))
+
 FEATURE_REQUEST_FACTORY(HelperFeatureEnum::DnnlUtils_engine_ext,
                         ASSIGNABLE_FACTORY(MEMBER_CALL_FACTORY_ENTRY(
                             "cudnnCreate", DEREF(ARG_WC(0)), false,

clang/lib/DPCT/APINames_cuDNN.inc

@@ -94,7 +94,7 @@ ENTRY(cudnnGetTensor4dDescriptor, cudnnGetTensor4dDescriptor, true, NO_FLAG, P4,
 ENTRY(cudnnGetTensorNdDescriptor, cudnnGetTensorNdDescriptor, true, NO_FLAG, P4, "Successful")
 ENTRY(cudnnGetTensorSizeInBytes, cudnnGetTensorSizeInBytes, true, NO_FLAG, P4, "Successful")
 ENTRY(cudnnGetTensorTransformDescriptor, cudnnGetTensorTransformDescriptor, false, NO_FLAG, P4, "comment")
-ENTRY(cudnnGetVersion, cudnnGetVersion, false, NO_FLAG, P4, "comment")
+ENTRY(cudnnGetVersion, cudnnGetVersion, true, NO_FLAG, P4, "comment")
 ENTRY(cudnnInitTransformDest, cudnnInitTransformDest, false, NO_FLAG, P4, "comment")
 ENTRY(cudnnLRNCrossChannelForward, cudnnLRNCrossChannelForward, true, NO_FLAG, P4, "Successful")
 ENTRY(cudnnNormalizationForwardInference, cudnnNormalizationForwardInference, true, NO_FLAG, P4, "comment")

clang/lib/DPCT/DNNAPIMigration.cpp

@@ -179,7 +179,7 @@ void CuDNNAPIRule::registerMatcher(ast_matchers::MatchFinder &MF) {
         "cudnnCreateDropoutDescriptor", "cudnnSetDropoutDescriptor",
         "cudnnGetDropoutDescriptor", "cudnnDropoutGetReserveSpaceSize",
         "cudnnRestoreDropoutDescriptor", "cudnnDropoutForward", "cudnnDropoutBackward",
-        "cudnnDestroyDropoutDescriptor");
+        "cudnnDestroyDropoutDescriptor", "cudnnGetVersion");
   };
 
   MF.addMatcher(

clang/runtime/dpct-rt/include/dnnl_utils.hpp.inc

@@ -38,6 +38,15 @@
 
 namespace dpct {
 namespace dnnl {
+// DPCT_LABEL_BEGIN|get_version|dpct::dnnl
+// DPCT_DEPENDENCY_EMPTY
+// DPCT_CODE
+/// Get concatenated library version as an integer.
+static inline size_t get_version() {
+  const ::dnnl::version_t *ver = ::dnnl::version();
+  return ver->major * 1000 + ver->minor * 100 + ver->patch;
+}
+// DPCT_LABEL_END
 class engine_ext;
 typedef oneapi::mkl::rng::philox4x32x10 rng_engine_t;
 // DPCT_LABEL_BEGIN|memory_format_tag|dpct::dnnl

clang/test/dpct/dnn/version.cu

@@ -0,0 +1,14 @@
+// RUN: dpct -in-root %S -out-root %T/version %S/version.cu --cuda-include-path="%cuda-path/include" -- -std=c++14 -x cuda --cuda-host-only
+// RUN: FileCheck --input-file %T/version/version.dp.cpp --match-full-lines %s
+#include <cuda_runtime.h>
+#include <cudnn.h>
+#include <iostream>
+#include <vector>
+
+
+int main() {
+    // CHECK: size_t version = dpct::dnnl::get_version();
+    size_t version = cudnnGetVersion();
+
+    return 0;
+}

clang/test/dpct/helper_files_ref/include/dnnl_utils.hpp

@@ -26,6 +26,11 @@
 
 namespace dpct {
 namespace dnnl {
+/// Get concatenated library version as an integer.
+static inline size_t get_version() {
+  const ::dnnl::version_t *ver = ::dnnl::version();
+  return ver->major * 1000 + ver->minor * 100 + ver->patch;
+}
 class engine_ext;
 typedef oneapi::mkl::rng::philox4x32x10 rng_engine_t;
 /// An enum class representing memory layout. Used by

clang/test/dpct/test_api_level/DnnlUtils/api_test28.cu

@@ -0,0 +1,20 @@
+// RUN: dpct --format-range=none --use-custom-helper=api -out-root %T/DnnlUtils/api_test28_out %s --cuda-include-path="%cuda-path/include" -- -x cuda --cuda-host-only
+// RUN: grep "IsCalled" %T/DnnlUtils/api_test28_out/MainSourceFiles.yaml | wc -l > %T/DnnlUtils/api_test28_out/count.txt
+// RUN: FileCheck --input-file %T/DnnlUtils/api_test28_out/count.txt --match-full-lines %s
+// RUN: rm -rf %T/DnnlUtils/api_test28_out
+
+
+// CHECK: 2
+// TEST_FEATURE: DnnlUtils_get_version
+
+#include <cuda_runtime.h>
+#include <cudnn.h>
+#include <iostream>
+#include <vector>
+
+int main() {
+
+    size_t ver = cudnnGetVersion();
+    return 0;
+
+}