Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
63 commits
Select commit Hold shift + click to select a range
1cc4739
Fusing refactor
Mar 10, 2025
a2e4632
remove commented out test file
Mar 12, 2025
1485cdc
revert changes in keras mixin
Mar 12, 2025
e08676f
remove unneeded fusion tests packages hirarchy
Mar 12, 2025
bb8cf01
revert old tests that were commented out
Mar 12, 2025
0bb6f5a
fix wrong type hint in test_activation_weights_composition_substition
Mar 12, 2025
d7135d6
remove commented out code from runner
Mar 12, 2025
8f3b1d9
use internal graph for the final qat model instead of the fused graph
Mar 12, 2025
4cc1533
add comment for second moment correction about the fusing info correc…
Mar 12, 2025
3f6fe40
remove old fusing data from graph
Mar 12, 2025
2ea3f2a
update old pytorch tests
Mar 12, 2025
595f4bd
add check for graph type in torch model builder
Mar 13, 2025
a9ae9f9
add comments to fusing info
Mar 14, 2025
e1cad46
add comments to graph fuser and graph with metadata
Mar 14, 2025
43fbf1d
Set version for onnxruntime-extensions
Mar 14, 2025
5c16f03
adapt keras unit tests
Mar 14, 2025
7c52939
Revert "Set version for onnxruntime-extensions"
Mar 17, 2025
7619677
Merge remote-tracking branch 'origin/main' into refactor-fusing
Mar 17, 2025
8406b2c
fix comments in fusing info
Mar 17, 2025
fb0a30b
remove the deepcopy in get_all_fused_operations
Mar 17, 2025
ec583f1
verify fusing info is consistent when using graph fuser
Mar 18, 2025
44f2256
move function to disable activation quantization from fusing info to …
Mar 18, 2025
5544545
use prefic of op id as constant
Mar 18, 2025
6e7d170
pass only fusing patterns instead of entire fqc
Mar 18, 2025
1ff0cc9
use dataclass for FusingInfo
Mar 18, 2025
c1a413b
rename filter_fusing_patterns
Mar 24, 2025
ce79fc8
remove duplicate def of FusedLayerType
Mar 24, 2025
459394e
rename fusing mateadata wrapper
Mar 24, 2025
671c58a
fix old name of FusingMetadataWrapper in comments
Mar 24, 2025
34c2820
fix comments in fuse method of GraphFuser
Mar 24, 2025
0054571
rename fuse method in graph fuser
Mar 24, 2025
bd75774
save fusing info instead of multiple fetches
Mar 24, 2025
b1be93a
set the graph without the fusing metadata in the pytorch back2fw
Mar 24, 2025
a589333
add missing types in BaseGraphWithFusingMetadataTest
Mar 24, 2025
30e3645
adjust the test of activation-weight composition due to the fusion re…
Mar 24, 2025
e7ac1de
migrate old unittests
Mar 25, 2025
f5c0a07
tests with multiple successors/predecessors
Mar 26, 2025
7551d34
add test that checks the case of new fusion due to new node
Mar 26, 2025
98c14f5
add test that checks a valid graph change does not fail the validatio…
Mar 26, 2025
8ee3191
merge changes with main
Mar 26, 2025
d78ac46
move tests to a designated package
Mar 26, 2025
4c396ae
fix batchnorm_reconstruction due to replacing list with tuple in fus…
Mar 26, 2025
4580a98
use internal graph for mp for now
Mar 26, 2025
3ab79d9
fix missing import of removed test
Mar 26, 2025
63a7dff
fix wrong path in test_cfg_candidates_filter
Mar 26, 2025
c170f2a
run pytest before unittests
Mar 26, 2025
773c115
add missing license
Mar 26, 2025
d31c53c
extract minimal_cfg_options from minimal_tpc
Mar 26, 2025
106e694
fix new argument in test_cfg_candidates_filter
Mar 27, 2025
a16e47a
replace wrapper with embedding the fusing info in the graph
Apr 1, 2025
f485a36
rewrite the funsions in test_fusing_info more explicity
Apr 1, 2025
8a0334b
return fusion to test in test_activation_weights_composition_substitu…
Apr 1, 2025
68b560c
disable validation in tests of virtual graph
Apr 1, 2025
d097e77
fix tests
Apr 1, 2025
2011098
merge from main
Apr 1, 2025
48ab653
rename tests files to remove wrapper
Apr 1, 2025
e112c65
run pytests in keras before unittests
Apr 1, 2025
b640c45
add integration tests for GraphFuser
Apr 2, 2025
39dcce1
disable validation in keras function tests for virtual graph
Apr 2, 2025
57320b0
use tuple for the op cfg options instead of list
Apr 2, 2025
3031dba
fix tpc in tests where default options had more than one option
Apr 2, 2025
e81ebf5
extend fusing info tests
Apr 2, 2025
7000c8d
change skip_validation flag to protected
Apr 2, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 3 additions & 4 deletions .github/workflows/run_keras_tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -27,11 +27,10 @@ jobs:
pip install tensorflow==${{ inputs.tf-version }} sony-custom-layers
pip install pytest pytest-mock
pip check
- name: Run unittests
run: |
python -m unittest discover tests/keras_tests -v

- name: Run pytest
run: |
pytest tests_pytest/keras_tests
- name: Run unittests
run: |
python -m unittest discover tests/keras_tests -v

7 changes: 4 additions & 3 deletions .github/workflows/run_pytorch_tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -33,10 +33,11 @@ jobs:
pip install torch==${{ inputs.torch-version }} torchvision onnx onnxruntime "onnxruntime-extensions<0.14"
pip install pytest pytest-mock
pip check
- name: Run unittests
run: |
python -m unittest discover tests/pytorch_tests -v
- name: Run pytest
run: |
pytest tests_pytest/pytorch_tests
- name: Run unittests
run: |
python -m unittest discover tests/pytorch_tests -v


Loading