Skip to content

Commit 7a08405

Browse files
reuvenperetzreuvenp
andauthored
Refactor fusing (#1386)
This commit introduces handling graph fusion by encapsulating fusion metadata in the graph. This ensures that after access to the graph that modify the graph, a validation check is performed to verify that the fusion information remains consistent and that no modifications have introduced inconsistencies. Additionally, the fusion-related logic has been refactored into a new class called GraphFuser, which takes the graph along with its fusion metadata and creates a new graph where fused operations are represented as single nodes. --------- Co-authored-by: reuvenp <reuvenp@altair-semi.com>
1 parent 516061b commit 7a08405

31 files changed

+2211
-785
lines changed

.github/workflows/run_keras_tests.yml

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,11 +27,10 @@ jobs:
2727
pip install tensorflow==${{ inputs.tf-version }} sony-custom-layers
2828
pip install pytest pytest-mock
2929
pip check
30-
- name: Run unittests
31-
run: |
32-
python -m unittest discover tests/keras_tests -v
33-
3430
- name: Run pytest
3531
run: |
3632
pytest tests_pytest/keras_tests
33+
- name: Run unittests
34+
run: |
35+
python -m unittest discover tests/keras_tests -v
3736

.github/workflows/run_pytorch_tests.yml

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,10 +33,11 @@ jobs:
3333
pip install torch==${{ inputs.torch-version }} torchvision onnx onnxruntime "onnxruntime-extensions<0.14"
3434
pip install pytest pytest-mock
3535
pip check
36-
- name: Run unittests
37-
run: |
38-
python -m unittest discover tests/pytorch_tests -v
3936
- name: Run pytest
4037
run: |
4138
pytest tests_pytest/pytorch_tests
39+
- name: Run unittests
40+
run: |
41+
python -m unittest discover tests/pytorch_tests -v
42+
4243

0 commit comments

Comments
 (0)