From 8208343afde54a94b10785780c0d4d5dd0592ff2 Mon Sep 17 00:00:00 2001 From: "q.yao" Date: Tue, 9 Nov 2021 17:44:17 +0800 Subject: [PATCH] solve torch and cub conflict (#178) --- backend_ops/tensorrt/CMakeLists.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/backend_ops/tensorrt/CMakeLists.txt b/backend_ops/tensorrt/CMakeLists.txt index 55572cf01..f2fed46a6 100644 --- 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