Fix Cudnn finder

Signed-off-by: Gabriel Cuendet <[email protected]>

Author: gcuendet-cognex

cmake/Modules/FindcuDNN.cmake

@@ -182,56 +182,62 @@ if(cuDNN_INCLUDE_DIRS)
   endif()
 endif()
 
+# pytorch compatibility layer
+set(CUDNN_LIBRARY_PATH ${cuDNN_LINK_LIBRARY})
+set(CUDNN_INCLUDE_PATH ${cuDNN_INCLUDE_DIRS})
+
 find_package_handle_standard_args(cuDNN
   REQUIRED_VARS cuDNN_LINK_LIBRARY cuDNN_INCLUDE_DIRS
   VERSION_VAR   cuDNN_VERSION)
 
 mark_as_advanced(cuDNN_LINK_LIBRARY cuDNN_INCLUDE_DIRS cuDNN_DLL_LIBRARY)
 
 if(cuDNN_FOUND)
-  add_library(cuDNN::cuDNN SHARED IMPORTED)
-  if(WIN32)
-    set_target_properties(cuDNN::cuDNN
-      PROPERTIES
-      IMPORTED_LINK_INTERFACE_LANGUAGE "C"
-      INTERFACE_INCLUDE_DIRECTORIES "${cuDNN_INCLUDE_DIRS}"
-      IMPORTED_LOCATION "${cuDNN_DLL_LIBRARY}"
-      IMPORTED_IMPLIB "${cuDNN_LINK_LIBRARY}"
-    )
-  else(WIN32)
-    set_target_properties(cuDNN::cuDNN
-      PROPERTIES
-      IMPORTED_LINK_INTERFACE_LANGUAGE "C"
-      INTERFACE_INCLUDE_DIRECTORIES "${cuDNN_INCLUDE_DIRS}"
-      IMPORTED_LOCATION "${cuDNN_LINK_LIBRARY}"
-    )
-  endif(WIN32)
-  if(cuDNN_VERSION_MAJOR VERSION_GREATER 8 OR cuDNN_VERSION_MAJOR VERSION_EQUAL 8)
-    macro(create_cudnn_target cudnn_target_name)
-	  string(TOUPPER ${cudnn_target_name} target_infix)
-	  add_library(cuDNN::${cudnn_target_name} SHARED IMPORTED)
-	  if(WIN32)
-        set_target_properties(cuDNN::${cudnn_target_name}
-          PROPERTIES
-          IMPORTED_LINK_INTERFACE_LANGUAGE "C"
-          INTERFACE_INCLUDE_DIRECTORIES "${cuDNN_INCLUDE_DIRS}"
-          IMPORTED_LOCATION "${cuDNN_${target_infix}_DLL_LIBRARY}"
-          IMPORTED_IMPLIB "${cuDNN_${target_infix}_LINK_LIBRARY}"
-        )
-      else(WIN32)
+  if(NOT TARGET cuDNN::cuDNN)
+    add_library(cuDNN::cuDNN SHARED IMPORTED)
+    if(WIN32)
+      set_target_properties(cuDNN::cuDNN
+        PROPERTIES
+        IMPORTED_LINK_INTERFACE_LANGUAGE "C"
+        INTERFACE_INCLUDE_DIRECTORIES "${cuDNN_INCLUDE_DIRS}"
+        IMPORTED_LOCATION "${cuDNN_DLL_LIBRARY}"
+        IMPORTED_IMPLIB "${cuDNN_LINK_LIBRARY}"
+      )
+    else(WIN32)
+      set_target_properties(cuDNN::cuDNN
+        PROPERTIES
+        IMPORTED_LINK_INTERFACE_LANGUAGE "C"
+        INTERFACE_INCLUDE_DIRECTORIES "${cuDNN_INCLUDE_DIRS}"
+        IMPORTED_LOCATION "${cuDNN_LINK_LIBRARY}"
+      )
+    endif(WIN32)
+    if(cuDNN_VERSION_MAJOR VERSION_GREATER 8 OR cuDNN_VERSION_MAJOR VERSION_EQUAL 8)
+      macro(create_cudnn_target cudnn_target_name)
+        string(TOUPPER ${cudnn_target_name} target_infix)
+        add_library(cuDNN::${cudnn_target_name} SHARED IMPORTED)
+        if(WIN32)
+          set_target_properties(cuDNN::${cudnn_target_name}
+            PROPERTIES
+            IMPORTED_LINK_INTERFACE_LANGUAGE "C"
+            INTERFACE_INCLUDE_DIRECTORIES "${cuDNN_INCLUDE_DIRS}"
+            IMPORTED_LOCATION "${cuDNN_${target_infix}_DLL_LIBRARY}"
+            IMPORTED_IMPLIB "${cuDNN_${target_infix}_LINK_LIBRARY}"
+          )
+        else(WIN32)
           set_target_properties(cuDNN::${cudnn_target_name}
             PROPERTIES
             IMPORTED_LINK_INTERFACE_LANGUAGE "C"
             INTERFACE_INCLUDE_DIRECTORIES "${cuDNN_INCLUDE_DIRS}"
             IMPORTED_LOCATION "${cuDNN_${target_infix}_LINK_LIBRARY}"
           )
-      endif(WIN32)
-	endmacro()
-	create_cudnn_target(adv_infer)
-	create_cudnn_target(adv_train)
-	create_cudnn_target(cnn_infer)
-	create_cudnn_target(cnn_train)
-	create_cudnn_target(ops_infer)
-	create_cudnn_target(ops_train)
-  endif()
-endif(cuDNN_FOUND)
+        endif(WIN32)
+      endmacro()
+      create_cudnn_target(adv_infer)
+      create_cudnn_target(adv_train)
+      create_cudnn_target(cnn_infer)
+      create_cudnn_target(cnn_train)
+      create_cudnn_target(ops_infer)
+      create_cudnn_target(ops_train)
+    endif()
+  endif(NOT TARGET cuDNN::cuDNN)
+endif(cuDNN_FOUND)