solve torch and cub conflict (open-mmlab#178)

humu789 · Nov 9, 2021 · 8208343 · 8208343
1 parent 391ddeb
commit 8208343
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/backend_ops/tensorrt/CMakeLists.txt b/backend_ops/tensorrt/CMakeLists.txt
@@ -24,6 +24,7 @@ else()
   set(CMAKE_CUDA_HOST_COMPILER ${CMAKE_CXX_COMPILER})
   set(CUDA_NVCC_FLAGS
       "${CUDA_NVCC_FLAGS} -Xcompiler=-fPIC,-Wall,-fvisibility=hidden")
+  set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -Xcompiler=-fno-gnu-unique")
 endif()
 
 # set virtual compute architecture and real ones