build: nvidia: fix install dependencies (#2768)

Author: sgeor255

cmake/config.cmake.in

@@ -21,6 +21,8 @@ set(DNNL_GPU_RUNTIME "@DNNL_GPU_RUNTIME@")
 
 set(DNNL_BLAS_VENDOR "@DNNL_BLAS_VENDOR@")
 
+set(DNNL_GPU_VENDOR "@DNNL_GPU_VENDOR@")
+
 if(DNNL_CPU_THREADING_RUNTIME STREQUAL "TBB")
     # Try to find TBB using a TBB-provided CMake config file.
     find_package(TBB QUIET COMPONENTS tbb)
@@ -62,6 +64,14 @@ check_required_components("@LIB_PACKAGE_NAME@")
 
 if(DNNL_CPU_RUNTIME STREQUAL "SYCL" OR DNNL_CPU_RUNTIME STREQUAL "DPCPP" OR
    DNNL_GPU_RUNTIME STREQUAL "SYCL" OR DNNL_GPU_RUNTIME STREQUAL "DPCPP")
+    if(DNNL_GPU_VENDOR STREQUAL "NVIDIA")
+        set(DNNL_ORIGINAL_CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH})
+        list(INSERT CMAKE_MODULE_PATH 0 ${PACKAGE_PREFIX_DIR}/@LIB_CONFIG_INSTALL_DIR@)
+        find_package(cuDNN REQUIRED)
+        find_package(cuBLAS REQUIRED)
+        find_package(cublasLt REQUIRED)
+        set(CMAKE_MODULE_PATH ${DNNL_ORIGINAL_CMAKE_MODULE_PATH})
+    endif()
     set(DNNL_COMPILE_FLAGS "-fsycl")
     @HANDLE_BUNDLE_DEBUG_SYCL_CONFIGURATION@
 endif()

src/CMakeLists.txt

@@ -305,6 +305,14 @@ if(DNNL_BLAS_VENDOR STREQUAL "ACCELERATE")
     DESTINATION ${LIB_CONFIG_INSTALL_DIR})
 endif()
 
+if(DNNL_SYCL_CUDA)
+    install(FILES
+        "../cmake/FindcuBLAS.cmake"
+        "../cmake/FindcublasLt.cmake"
+        "../cmake/FindcuDNN.cmake"
+        DESTINATION ${LIB_CONFIG_INSTALL_DIR})
+endif()
+
 # On Windows we need to add dnnl.dll path to CTESTCONFIG_PATH which is later
 # passed to ctest and Visual Studio solutions
 if(WIN32)