Skip to content

GitLab

You need to sign in or sign up before continuing.
Unverified Commit 48f44b69 authored by Ming-Hsuan-Tu's avatar Ming-Hsuan-Tu Committed by GitHub
Browse files

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

* fix tensorrt test failed with pytorch 1.8+

* add comment
parent 222f3807
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
setup.py
View file @ 48f44b69
...@@ -161,6 +161,9 @@ def get_extensions(): ...@@ -161,6 +161,9 @@ def get_extensions():
define_macros += [('MMCV_WITH_TRT', None)] define_macros += [('MMCV_WITH_TRT', None)]
cuda_args = os.getenv('MMCV_CUDA_ARGS') cuda_args = os.getenv('MMCV_CUDA_ARGS')
extra_compile_args['nvcc'] = [cuda_args] if cuda_args else [] 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) library_dirs += library_paths(cuda=True)
from setuptools import Extension from setuptools import Extension
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment