Level of support for Transformer ML Models in torch MLIR

Incubator torch-mlir
1

Hello all,

I wanted to find out the level of support already built for supporting Transformer ML models in Torch MLIR. Is there a web site that lists which all of Transformer models (listed on 🤗 Transformers) have been tested to be handled correctly for PyTorch Torchscript to Torch MLIR (or TOSA MLIR) conversion and subsequent lowering to Reference MLIR CPU runner ?

Thanks for any guidance on this.
Kumar

2

@powderluv

3

Here is a list of models that Nod has been able to run using Torch-MLIR: SHARK/README.md at main · nod-ai/SHARK · GitHub

1 Like