Skip to content

Commit

Permalink
[ONNX] Enable opset 13 ops (pytorch#49612)
Browse files Browse the repository at this point in the history
* Enable opset 13 ORT tests

* Update test.sh

* Set environ var

* Update test.sh

* Enabling more ops for opset 13

* change master to main

* Update symbolic_opset13.py

* Flake 8 fix

* [ONNX] Support onnx if/loop sequence output in opset 13 - (pytorch#49270)

* Symbolic function for torch.square (pytorch#49446)

* Clean up tests

* Exclude more tests

* Trigge build

* [ONNX] Support onnx if/loop sequence output in opset 13 - (pytorch#49270)

* Symbolic function for torch.square (pytorch#49446)

* update ORT version

* disable more tests

* clean up

* flake8

* Disable TV tests

* Update test_pytorch_onnx_onnxruntime.py

Co-authored-by: Bowen Bao <bowbao@microsoft.com>
Co-authored-by: David Fan <30608893+jiafatom@users.noreply.github.com>
  • Loading branch information
3 people authored and Spandan Tiwari committed Jan 8, 2021
1 parent 6e5629d commit 672d0a0
Show file tree
Hide file tree
Showing 7 changed files with 248 additions and 17 deletions.
2 changes: 1 addition & 1 deletion scripts/onnx/test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ if [[ "$BUILD_ENVIRONMENT" == *ort_test1* ]]; then
fi
if [[ "$BUILD_ENVIRONMENT" == *ort_test2* ]]; then
# Update the loop for new opsets
for i in $(seq 10 12); do
for i in $(seq 10 13); do
pytest "${args[@]}" \
"$top_dir/test/onnx/test_pytorch_onnx_onnxruntime.py::TestONNXRuntime_opset$i"
done
Expand Down
Loading

0 comments on commit 672d0a0

Please sign in to comment.