Skip to content

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

Open
@peri044

Description

@peri044

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

Metadata

Metadata

Assignees

Labels

StoryIssues proposing a new Story

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions