From 4efef5bfca4e43bcc23d7707a73e0879c931994a Mon Sep 17 00:00:00 2001 From: David Date: Tue, 8 Dec 2020 12:07:24 -0800 Subject: [PATCH 1/2] Enable ONNX test in circle CI --- .circleci/config.yml | 3 +++ .circleci/config.yml.in | 3 +++ 2 files changed, 6 insertions(+) diff --git a/.circleci/config.yml b/.circleci/config.yml index c23c5fc5095..906adeac2b7 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -144,6 +144,9 @@ jobs: # need to install torchvision dependencies due to transitive imports pip install --user --progress-bar off --editable . python test/test_hub.py + pip install --user onnx + pip install --user -i https://test.pypi.org/simple/ ort-nightly==1.5.2.dev202012031 + python test/test_onnx.py binary_linux_wheel: <<: *binary_common diff --git a/.circleci/config.yml.in b/.circleci/config.yml.in index 70eaff7f815..1ec188183cc 100644 --- a/.circleci/config.yml.in +++ b/.circleci/config.yml.in @@ -144,6 +144,9 @@ jobs: # need to install torchvision dependencies due to transitive imports pip install --user --progress-bar off --editable . python test/test_hub.py + pip install --user onnx + pip install --user -i https://test.pypi.org/simple/ ort-nightly==1.5.2.dev202012031 + python test/test_onnx.py binary_linux_wheel: <<: *binary_common From ef3ebbeb0f5f6104010c5574726a520a23dfde63 Mon Sep 17 00:00:00 2001 From: David Date: Tue, 8 Dec 2020 12:07:24 -0800 Subject: [PATCH 2/2] Enable ONNX test in circle CI --- .circleci/config.yml | 13 +++++++++++++ .circleci/config.yml.in | 13 +++++++++++++ 2 files changed, 26 insertions(+) diff --git a/.circleci/config.yml b/.circleci/config.yml index c23c5fc5095..35503c031f6 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -145,6 +145,17 @@ jobs: pip install --user --progress-bar off --editable . python test/test_hub.py + torch_onnx_test: + docker: + - image: circleci/python:3.7 + steps: + - checkout + - run: + command: | + pip install --user onnx + pip install --user -i https://test.pypi.org/simple/ ort-nightly==1.5.2.dev202012031 + python test/test_onnx.py + binary_linux_wheel: <<: *binary_common docker: @@ -1074,6 +1085,7 @@ workflows: - python_type_check - clang_format - torchhub_test + - torch_onnx_test unittest: jobs: @@ -1189,6 +1201,7 @@ workflows: - python_type_check - clang_format - torchhub_test + - torch_onnx_test - binary_linux_wheel: cu_version: cpu filters: diff --git a/.circleci/config.yml.in b/.circleci/config.yml.in index 70eaff7f815..101be49f2db 100644 --- a/.circleci/config.yml.in +++ b/.circleci/config.yml.in @@ -145,6 +145,17 @@ jobs: pip install --user --progress-bar off --editable . python test/test_hub.py + torch_onnx_test: + docker: + - image: circleci/python:3.7 + steps: + - checkout + - run: + command: | + pip install --user onnx + pip install --user -i https://test.pypi.org/simple/ ort-nightly==1.5.2.dev202012031 + python test/test_onnx.py + binary_linux_wheel: <<: *binary_common docker: @@ -700,6 +711,7 @@ workflows: - python_type_check - clang_format - torchhub_test + - torch_onnx_test unittest: jobs: @@ -717,6 +729,7 @@ workflows: - python_type_check - clang_format - torchhub_test + - torch_onnx_test {{ build_workflows(prefix="nightly_", filter_branch="nightly", upload=True) }} docker_build: triggers: