IntenseWebs/openvino
3
0 Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GHA] Add torch.compile layer tests to GHA (#20655)

Browse Source 
* [GHA] Add torch.compile layer tests to GHA

* Update pytorch_layer_test_class.py
This commit is contained in:
Maxim Vafin authored and GitHub committed 2023-10-30 17:23:02 +04:00
1 parent 56cb121e8a
commit 439b7e57c7
1 file changed
+17 -1
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/linux.yml
+17 -1
View File
@@ -1018,7 +1018,23 @@ jobs:
run: python3 -m pytest ${LAYER_TESTS_INSTALL_DIR}/pytorch_tests -n logical -m precommit --junitxml=${INSTALL_TEST_DIR}/TEST-pytorch.xml
env:
TEST_DEVICE: CPU
TEST_PRECISION: FP16
TEST_PRECISION: FP32
- name: PyTorch torch.compile TORCHFX Layer Tests
run: |
python3 -m pytest ${LAYER_TESTS_INSTALL_DIR}/pytorch_tests -m precommit_fx_backend --junitxml=${INSTALL_TEST_DIR}/TEST-pytorch.xml
env:
TEST_DEVICE: CPU
TEST_PRECISION: FP32
PYTORCH_TRACING_MODE: TORCHFX
- name: PyTorch torch.compile TORCHSCRIPT Layer Tests
run: |
python3 -m pytest ${LAYER_TESTS_INSTALL_DIR}/pytorch_tests -m precommit_ts_backend --junitxml=${INSTALL_TEST_DIR}/TEST-pytorch.xml
env:
TEST_DEVICE: CPU
TEST_PRECISION: FP32
PYTORCH_TRACING_MODE: TORCHSCRIPT
- name: ONNX Layer Tests
run: |