• q.yao's avatar
    [Feature]: Add custom operators support for TensorRT in mmcv (#686) · 0de9e149
    q.yao authored
    
    
    * start trt plugin prototype
    
    * Add test module, modify roialign convertor
    
    * finish roi_align trt plugin
    
    * fix conflict of RoiAlign and MMCVRoiAlign
    
    * fix for lint
    
    * fix test tensorrt module
    
    * test_tensorrt move import to test func
    
    * add except error type
    
    * add tensorrt to setup.cfg
    
    * code format with yapf
    
    * fix for clang-format
    
    * move tensorrt_utils to mmcv/tensorrt, add comments, better test module
    
    * fix line endings, docformatter
    
    * isort init, remove trailing whitespace
    
    * add except type
    
    * fix setup.py
    
    * put import extension inside trt setup
    
    * change c++ guard, update pytest script, better setup, etc
    
    * sort import with isort
    
    * sort import with isort
    
    * move init of plugin lib to init_plugins.py
    
    * resolve format and add test dependency: tensorrt
    
    * tensorrt should be installed from source not from pypi
    
    * update naming style and input check
    
    * resolve lint error
    Co-authored-by: default avatarmaningsheng <maningsheng@sensetime.com>
    0de9e149
setup.py 10.9 KB