Skip to content

Commit ae831e6

Browse files
authored
Merge branch 'main' into justinchu/ser-initializer
2 parents 5156668 + 159e5bc commit ae831e6

File tree

251 files changed

+6643
-1378
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

251 files changed

+6643
-1378
lines changed

.github/workflows/main.yaml

Lines changed: 25 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -99,12 +99,32 @@ jobs:
9999
with:
100100
name: Error reports (${{ matrix.name }}-${{ matrix.os }})
101101
path: error_reports
102-
- name: Upload IR profiling results
103-
if: matrix.name == 'py311' || matrix.name == 'py311-onnx-weekly'
104-
uses: actions/upload-artifact@v3
102+
103+
dort:
104+
strategy:
105+
fail-fast: false
106+
matrix:
107+
os: [ubuntu-latest]
108+
transformers: ["4.37.2", "4.41.2"]
109+
torch: ["release", "nightly"]
110+
python_version: ["3.11"]
111+
nox-tag: ["test-dort"]
112+
name:
113+
- dort
114+
runs-on: ${{ matrix.os }}
115+
steps:
116+
- uses: actions/checkout@v4
117+
- name: Setup Python ${{ matrix.python_version }}
118+
uses: actions/setup-python@v5
105119
with:
106-
name: IR profiling results
107-
path: tests/ir/serde_test_profiles
120+
python-version: ${{ matrix.python_version }}
121+
- name: Install nox
122+
run: python -m pip install nox
123+
- name: Pull Test Data
124+
run: git lfs pull
125+
- run: |
126+
nox -t ${{ matrix.nox-tag }} --forcecolor -- ${{ matrix.torch }} ${{ matrix.transformers }}
127+
name: Run tests
108128
109129
build_docs:
110130
strategy:

.gitignore

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ coverage.xml
4141
.pytest_cache/
4242
cover/
4343
test-output.xml
44+
*.sarif
4445

4546
# Sphinx documentation
4647
docs/_build/
@@ -93,6 +94,8 @@ dmypy.json
9394

9495
# Generated files
9596
*.onnx
97+
*.csv
98+
*.xlsx
9699
!testdata/**/*.onnx
97100
*.onnxlib
98101
**/onnx_backend_test_code/**

README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@ models using a subset of Python. ONNX Script is:
1818
Note however that ONNX Script does **not** intend to support the entirety
1919
of the Python language.
2020

21+
Website: [https://onnxscript.ai/](https://onnxscript.ai/)
22+
2123
## Design Overview
2224

2325
ONNX Script provides a few major capabilities for authoring and debugging

docs/api/index.md

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,17 @@
11
# API
22

3+
## Author Models
4+
35
```{toctree}
46
decorator
57
opsets
68
converter
79
values
810
```
11+
12+
## Tests and Tools
13+
14+
```{toctree}
15+
testing
16+
tools
17+
```

docs/api/testing.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
# Testing
2+
3+
```{eval-rst}
4+
.. automodule:: onnxscript.testing
5+
:members:
6+
```

docs/api/tools.md

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
# Tools
2+
3+
## Transformers Models
4+
5+
```{eval-rst}
6+
.. autofunction:: onnxscript.tools.transformers_models.get_model_and_inputs
7+
```
8+
9+
```{eval-rst}
10+
.. autofunction:: onnxscript.tools.transformers_models.phi.get_phi_model_from_config
11+
```
12+
13+
```{eval-rst}
14+
.. autofunction:: onnxscript.tools.transformers_models.phi3.get_phi3_model_from_config
15+
```
16+
17+
```{eval-rst}
18+
.. autofunction:: onnxscript.tools.transformers_models.llama.get_llama_model_from_config
19+
```

docs/conf.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,9 @@
1-
# Configuration file for the Sphinx documentation builder.
2-
# To run the documentation: python -m sphinx docs dist/html
1+
# Copyright (c) Microsoft Corporation.
2+
# Licensed under the MIT License.
3+
"""Configuration file for the Sphinx documentation builder.
4+
5+
To run the documentation: python -m sphinx docs dist/html
6+
"""
37

48
import os
59
import re

docs/examples/01_plot_selu.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
# Copyright (c) Microsoft Corporation.
2+
# Licensed under the MIT License.
13
"""
24
Generating a FunctionProto
35
==========================

docs/examples/02_plot_square_loss.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
# Copyright (c) Microsoft Corporation.
2+
# Licensed under the MIT License.
13
"""
24
Generating a ModelProto
35
=======================

docs/examples/03_export_lib.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
# Copyright (c) Microsoft Corporation.
2+
# Licensed under the MIT License.
13
"""
24
Generating a LibProto
35
=====================

0 commit comments

Comments
 (0)