From 48f44b697a3510803a252c78538a9064458995a5 Mon Sep 17 00:00:00 2001
From: Ming-Hsuan-Tu <qrnnis2623891@gmail.com>
Date: Fri, 10 Dec 2021 19:41:37 +0800
Subject: [PATCH] fix tensorrt test failed with pytorch 1.8+ (#1464)

* fix tensorrt test failed with pytorch 1.8+

* add comment
---
 setup.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/setup.py b/setup.py
index 1284f9f5d7..ba62ea2671 100644
--- a/setup.py
+++ b/setup.py
@@ -161,6 +161,9 @@ def get_extensions():
         define_macros += [('MMCV_WITH_TRT', None)]
         cuda_args = os.getenv('MMCV_CUDA_ARGS')
         extra_compile_args['nvcc'] = [cuda_args] if cuda_args else []
+        # prevent cub/thrust conflict with other python library
+        # More context See issues #1454
+        extra_compile_args['nvcc'] += ['-Xcompiler=-fno-gnu-unique']
         library_dirs += library_paths(cuda=True)
 
         from setuptools import Extension