docs.rs failed to build easy-tensorrt-core-0.3.0
Please check the build logs for more information.
See Builds for ideas on how to fix a failed build, or Metadata for how to configure docs.rs builds.
If you believe this is docs.rs' fault, open an issue.
Please check the build logs for more information.
See Builds for ideas on how to fix a failed build, or Metadata for how to configure docs.rs builds.
If you believe this is docs.rs' fault, open an issue.
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.