Amirstan_plugin

Amirstan plugin contain some useful tensorrt plugin. These plugins are used to support some other project such as

https://github.com/grimoire/torch2trt_dynamic

https://github.com/grimoire/mmdetection-to-tensorrt

Requirement

• Tensorrt >= 7.0.0.11
• cub >= 1.8.0

Installation

config CUB_ROOT_DIR / TENSORRT_ROOT or other library in CMakeLists.txt

git clone https://github.com/grimoire/amirstan_plugin.git
cd amirstan_plugin
mkdir build
cd build
cmake ..
make -j10

set the envoirment variable(in ~/.bashrc):

export AMIRSTAN_LIBRARY_PATH=<amirstan_plugin_root>/build/lib