Skip to content

Commit

Permalink
Optimize the model analysis weekly pipeline (#874)
Browse files Browse the repository at this point in the history
The PR will optimise the model analysis weekly pipeline - #761 

The current model_analysis script will generate the unique op tests and
running the unique op configuration test model wise instead collect the
unique op test configuration from all the models and then extract unique
op configuration test across all the models which will avoid running
same configuration that are present in multiple models again and again.

For example, if we have add op with operands shape of (1, 32) and (1,
32) in resnet and mobilenet, will run the test only once and populate
the test results in both resnet and mobilenet
  • Loading branch information
chandrasekaranpradeep authored Dec 24, 2024
1 parent b130215 commit 605e6c9
Show file tree
Hide file tree
Showing 4 changed files with 573 additions and 334 deletions.
2 changes: 1 addition & 1 deletion forge/forge/python_codegen.py
Original file line number Diff line number Diff line change
Expand Up @@ -1085,7 +1085,7 @@ def write_pytest_function(
self.wl("")
self.wl("compiled_model = compile(framework_model, sample_inputs=inputs)")
self.wl("")
self.wl("verify(inputs, framework_model, compiled_model, VerifyConfig(verify_allclose=False))")
self.wl("verify(inputs, framework_model, compiled_model)")
self.wl("")
self.wl("")
self.indent -= 1
Expand Down
Loading

0 comments on commit 605e6c9

Please sign in to comment.