Fix cub (open-mmlab#312)

* fix cub path * use cmake source dir instead
humu789 · Dec 20, 2021 · 547a160 · 547a160
1 parent d9b23bf
commit 547a160
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/csrc/backend_ops/tensorrt/CMakeLists.txt b/csrc/backend_ops/tensorrt/CMakeLists.txt
@@ -12,7 +12,7 @@ set_targets(${PROJECT_NAME} BACKEND_OPS_OBJ BACKEND_OPS_STATIC BACKEND_OPS_MODUL
 # cub
 if (NOT DEFINED CUB_ROOT_DIR)
     if (CUDA_VERSION VERSION_LESS 11.0)
-        set(CUB_ROOT_DIR "${PROJECT_SOURCE_DIR}/third_party/cub")
+        set(CUB_ROOT_DIR "${CMAKE_SOURCE_DIR}/third_party/cub")
     endif()
 endif ()