Skip to content

✨[Feature]Implement the fx2trt coverters using the torch2trt converters #1657

Closed
@apbose

Description

@apbose

Feature context.
This feature tracks the progress in implementing the fx2trt converters using the torch2trt converters

Task steps:

  1. Run the unit tests in TensorRT TensorRT/py/torch_tensorrt/fx/test/aten_op/*.py to study acc and aten trace and how they differ
    STATUS : done

  2. Run progressively the model list and start supporting all the ops in the models. The model list:
    squeezenet1_0
    mobilenet_v2
    inception_v3
    efficientnet_b0
    regnet_y_8gf
    STATUS: At present the test exists for resnet18 /TensorRT/py/torch_tensorrt/fx/test/tracer/test_resnet.py. Expand this for the above networks

  3. In the process, integrate the converters such that they are agnostic to the aten and acc tracers. So we would want to make the flow generic to fx2trt
    STATUS : In progress

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions