1 unstable release
new 0.3.0 | May 22, 2025 |
---|---|
0.2.0 |
|
#2 in #tensor-rt
33 downloads per month
365KB
967 lines
easy-tensorrt-core
A safe wrapper around easy-tensorrt-sys. Provide the core functions to build and run TensorRT engine.
Inspired by tensorrt-rs.
Major changes from tensorrt-rs:
- Replace cuda-rs with cudarc.
- Replace Tensor with AbstractTensor trait. Any tensor (e.g. tch::Tensor) that implements AbstractTensor can be used.
- Add init module to faciitate the initialization of tensorrt plugins and cuda context.
- Init engine with CuStream and avoid passing in another CuStream when inference.
Dependencies
~8–18MB
~231K SLoC