Skip to content

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

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
apbose opened this issue Feb 10, 2023 · 1 comment
Closed
Assignees
Labels
feature request New feature or request No Activity

Comments

@apbose
Copy link
Collaborator

apbose commented Feb 10, 2023

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

@github-actions
Copy link

This issue has not seen activity for 90 days, Remove stale label or comment or this will be closed in 10 days

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request New feature or request No Activity
Projects
None yet
Development

No branches or pull requests

2 participants