TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

📖 [Story] Implement support for serde of TRT Graph modules with torch.export serde API

Open peri044 opened this issue 2 years ago • 0 comments

TL;DR

torch.export has serde API and now supports custom object (in our case torch.classes.tensorrt.Engine) serialization.

Goal(s)

We should be able to serialize/deserialize the output of torch-trt compilation.

Tasks

### Tasks
- [ ] https://github.com/pytorch/TensorRT/issues/2258
- [ ] https://github.com/pytorch/TensorRT/issues/2259
- [ ] https://github.com/pytorch/TensorRT/issues/2260
- [ ] https://github.com/pytorch/TensorRT/issues/2261
- [ ] https://github.com/pytorch/TensorRT/issues/2270
- [ ] https://github.com/pytorch/TensorRT/issues/2478
- [ ] https://github.com/pytorch/TensorRT/issues/2479
- [ ] https://github.com/pytorch/TensorRT/issues/2341
- [ ] https://github.com/pytorch/TensorRT/issues/2553
- [ ] https://github.com/pytorch/TensorRT/issues/2576

Additional context

peri044 avatar Aug 24 '23 17:08 peri044