fix tensorrt test failed with pytorch 1.8+ (#1464)

* fix tensorrt test failed with pytorch 1.8+ * add comment

fix tensorrt test failed with pytorch 1.8+ (#1464)
* fix tensorrt test failed with pytorch 1.8+ * add comment
48f44b69 · Ming-Hsuan-Tu · GitHub · 222f3807 · 48f44b69
Unverified Commit 48f44b69 authored Dec 10, 2021 by Ming-Hsuan-Tu Committed by GitHub Dec 10, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

setup.py setup.py +3 -0

No files found.
--- 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