https://github.com/justinchuby/torch-onnx-op-matrix

Continuously checks torch.onnx op support status
https://github.com/justinchuby/torch-onnx-op-matrix

onnx pytorch torchscript

Last synced: about 1 month ago
JSON representation

Continuously checks torch.onnx op support status

• Host: GitHub
• URL: https://github.com/justinchuby/torch-onnx-op-matrix
• Owner: justinchuby
• License: mit
• Created: 2022-10-08T00:12:09.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2025-04-21T22:15:00.000Z (about 1 month ago)
• Last Synced: 2025-04-21T23:19:52.426Z (about 1 month ago)
• Topics: onnx, pytorch, torchscript
• Language: Python
• Homepage: https://justinchuby.com/torch-onnx-op-matrix
• Size: 2.61 MB
• Stars: 1
• Watchers: 3
• Forks: 0
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# `torch.onnx` op support matrix

[![CI](https://github.com/justinchuby/torch-onnx-op-matrix/actions/workflows/main.yml/badge.svg)](https://github.com/justinchuby/torch-onnx-op-matrix/actions/workflows/main.yml)

[![Test](https://github.com/justinchuby/torch-onnx-op-matrix/actions/workflows/test.yml/badge.svg)](https://github.com/justinchuby/torch-onnx-op-matrix/actions/workflows/test.yml)

[![CoreML Tools](https://github.com/justinchuby/torch-onnx-op-matrix/actions/workflows/coremltools.yml/badge.svg)](https://github.com/justinchuby/torch-onnx-op-matrix/actions/workflows/coremltools.yml)

[![Deployment](https://github.com/justinchuby/torch-onnx-op-matrix/actions/workflows/deploy-react.yml/badge.svg)](https://github.com/justinchuby/torch-onnx-op-matrix/actions/workflows/deploy-react.yml)

`^` Obtain the data here under `artifacts`

Tool for continuously checking `torch.onnx` operator support status. The tool tests the `torch.onnx.{export, dynamo_export}` exporter on all {torch operators, dtype} combinations and reports any errors.

Additional tools also test running `torch.jit` on the ops, as well as running the `coremltools` exporter for comparison.

## Development

Visualization under `op-vis/`

```sh

npm install --legacy-peer-deps

```