From a835e91f17ed9914bf1f86a08df88d93d0f0cfbf Mon Sep 17 00:00:00 2001
From: Jun Tseng <6501202@qq.com>
Date: Mon, 23 Aug 2021 02:43:31 +0800
Subject: [PATCH] Remove unnecessary dependency for CUDA 11.x

According to https://github.com/NVIDIA/cub, the CUB is included in the CUDA Toolkit since 11.0 EA; As a result, there is no need to depend on a third-party CUB for CUDA 11.x.

Signed-off-by: Jun Tseng <6501202@qq.com>
---
 CMakeLists.txt | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index ba3af18fd..6cc4082ab 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -83,7 +83,6 @@ endif()
 set(DEFAULT_CUDA_VERSION 11.3.1)
 set(DEFAULT_CUDNN_VERSION 8.2)
 set(DEFAULT_PROTOBUF_VERSION 3.0.0)
-set(DEFAULT_CUB_VERSION 1.8.0)
 
 # Dependency Version Resolution
 set_ifndef(CUDA_VERSION ${DEFAULT_CUDA_VERSION})
@@ -99,7 +98,9 @@ if (BUILD_PLUGINS OR BUILD_PARSERS)
     include(third_party/protobuf.cmake)
 endif()
 if(NOT CUB_ROOT_DIR)
+  if (CUDA_VERSION VERSION_LESS 11.0)
     set(CUB_ROOT_DIR ${CMAKE_CURRENT_SOURCE_DIR}/third_party/cub CACHE STRING "directory of CUB installation")
+  endif()
 endif()
 
 ## find_package(CUDA) is broken for cross-compilation. Enable CUDA language instead.