- CUDA: 11.6
- TENSORRT: TensorRT-8.6.1.6
mkdir build
cd build
cmake ..
make -j8
make installcd test
# 生成trt engine并对比trt算子结果与torch算子结果
python3 torch_matmul_to_trt.py
# 生成trt engine并对比trt算子结果与tf算子结果
python3 tf_matmul_to_trt.py-
Notifications
You must be signed in to change notification settings - Fork 1
zhaocc1106/my_trt_plugin
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
实现自己的tensorrt算子
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published