[gpuDNN] Add gpuDNN v0.1 library. (#291)

Author: feng-intel

.gitignore

@@ -29,6 +29,7 @@ dist/
 /3rdparty/cmdline
 /3rdparty/sentencepiece
 /3rdparty/xdnn
+/3rdparty/gpudnn
 
 
 # MLServer

CMakeLists.txt

@@ -116,6 +116,13 @@ link_directories(${CMAKE_SOURCE_DIR}/3rdparty/onednn/build/src)
 link_directories(${CMAKE_SOURCE_DIR}/3rdparty/xdnn)
 link_directories(${CMAKE_SOURCE_DIR}/3rdparty/mkl/lib)
 
+if(WITH_GPU)
+    include("cmake/gpudnn.cmake")
+    include_directories(${CMAKE_SOURCE_DIR}/3rdparty/gpudnn)
+    link_directories(${CMAKE_SOURCE_DIR}/3rdparty/gpudnn)
+    list(APPEND DEPEND_LIST "gpudnn_lib")
+endif()
+
 find_package(oneCCL REQUIRED)
 
 if(MPI_FOUND AND oneCCL_FOUND)
@@ -143,9 +150,15 @@ option(BUILD_WITH_SHARED_LIBS "Build with shared libraries" OFF)
 if(BUILD_WITH_SHARED_LIBS)
     message(STATUS "Notice: Building with shared libraries.")
     list(APPEND 3RDPART_LIB_LIST "xdnn")
+    if(WITH_GPU)
+        list(APPEND 3RDPART_LIB_LIST "gpudnn")
+    endif()
 else()
     message(STATUS "Notice: Building with static libraries.")
     list(APPEND 3RDPART_LIB_LIST "xdnn_static")
+    if(WITH_GPU)
+        list(APPEND 3RDPART_LIB_LIST "gpudnn_static")
+    endif()
 endif()
 
 # pipeline parallel feature

cmake/gpudnn.cmake

@@ -0,0 +1,38 @@
+# Copyright (c) 2024 Intel Corporation
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+
+cmake_minimum_required(VERSION 3.18)
+
+# Avoid warning about DOWNLOAD_EXTRACT_TIMESTAMP in CMake 3.24:
+if(${CMAKE_VERSION} VERSION_GREATER_EQUAL "3.24.0")
+    cmake_policy(SET CMP0135 NEW)
+endif()
+
+project(dependency NONE)
+
+include(ExternalProject)
+
+# cmake-format: off
+ExternalProject_Add(gpudnn_lib
+  URL               https://github.com/intel/xFasterTransformer/releases/download/gpuDNN/gpudnn_v0.1.tar.gz
+  URL_HASH          MD5=05b3554413e454ed027014e44a5c7fe4
+  TIMEOUT           60
+  SOURCE_DIR        ${CMAKE_SOURCE_DIR}/3rdparty/gpudnn
+  CONFIGURE_COMMAND ""
+  BUILD_COMMAND     ""
+  INSTALL_COMMAND   ""
+  TEST_COMMAND      ""
+)
+# cmake-format: on