Commit 7a08405
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
File tree
31 files changed
+2211
-785
lines changed- .github/workflows
- model_compression_toolkit/core
- common
- fusion
- graph
- mixed_precision
- quantization
- substitutions
- tests_pytest
- _fw_tests_common_base
- fusing
- _test_util
- common_tests/unit_tests/core
- keras_tests/integration_tests/core/fusion
- pytorch_tests/integration_tests/core/fusion
- tests
- keras_tests
- feature_networks_tests/feature_networks
- function_tests
- pytorch_tests
- function_tests
- model_tests/feature_models
31 files changed
+2211
-785
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
27 | 27 | | |
28 | 28 | | |
29 | 29 | | |
30 | | - | |
31 | | - | |
32 | | - | |
33 | | - | |
34 | 30 | | |
35 | 31 | | |
36 | 32 | | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
37 | 36 | | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
33 | 33 | | |
34 | 34 | | |
35 | 35 | | |
36 | | - | |
37 | | - | |
38 | | - | |
39 | 36 | | |
40 | 37 | | |
41 | 38 | | |
| 39 | + | |
| 40 | + | |
| 41 | + | |
| 42 | + | |
42 | 43 | | |
0 commit comments