https://github.com/zhaocc1106/my_trt_plugin
实现自己的tensorrt算子
https://github.com/zhaocc1106/my_trt_plugin
tensorrt tensorrt-plugins
Last synced: 2 months ago
JSON representation
实现自己的tensorrt算子
- Host: GitHub
- URL: https://github.com/zhaocc1106/my_trt_plugin
- Owner: zhaocc1106
- Created: 2023-07-21T05:00:27.000Z (almost 2 years ago)
- Default Branch: master
- Last Pushed: 2023-08-09T19:20:39.000Z (almost 2 years ago)
- Last Synced: 2023-08-09T20:58:10.547Z (almost 2 years ago)
- Topics: tensorrt, tensorrt-plugins
- Language: C++
- Homepage:
- Size: 32.2 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# 实现自己的tensorrt算子插件
## 环境
* CUDA: 11.6
* TENSORRT: TensorRT-8.6.1.6
## 构建
```bash
mkdir build
cd build
cmake ..
make -j8
make install
```
## 测试
### 自定义matmul算子
```bash
cd test
# 生成trt engine并对比trt算子结果与torch算子结果
python3 torch_matmul_to_trt.py
# 生成trt engine并对比trt算子结果与tf算子结果
python3 tf_matmul_to_trt.py
```