[Frontend] minimal implementation of TorchFx importer - Githubissues

tlc-pack / relax

Apache License 2.0

193 stars 58 forks source link

[Frontend] minimal implementation of TorchFx importer #383

Closed sunggg closed 1 year ago

sunggg commented 1 year ago

This PR implements TorchFx importer for the minimal set of torch operators to demonstrate fallback mechanism (tracked by https://github.com/tlc-pack/relax/issues/373).

List of Operators

Torch operators
- torch.add
- torch.mul
Python builtin operators
- operator.add
- operator.mul
- operator.getitem

This implementation is based on the implementation in mlc-ai repo.

cc. @MasterJH5574 @spectrometerHBH @junrushao @YuchenJin

sunggg commented 1 year ago

Closed to reflect the latest change in mlc-ai.