Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
### Ticket [Link to Github Issue](#14912) ### Problem description Currently, we have approximately `14,000` tests in `max_pool2d`. This large number is due to test parameters being generated using the Cartesian product of all possible values for each parameter, resulting in an excessive test count. As a consequence, these tests take a significant amount of time to run in the CI/CD pipeline, leading to inefficiencies. ### What's changed This MR refactors the test by adopting an approach from `tt-metal/tests/sweep_framework/sweeps/max_pool2d/short/max_pool2d_short_sweep.py`. The key idea is to reduce the number of test parameters by focusing only on those actually used in practice. Instead of generating a Cartesian product of all possible values, we collect parameters directly from models that use `max_pool2d`. This set of models includes YOLOv4, VGG, `ttnn_resnet`, SegFormer, and `functional_unet`. ### Checklist - [x] [All post commit](https://github.com/tenstorrent/tt-metal/actions/workflows/all-post-commit-workflows.yaml) CI passes - [ ] [Blackhole Post commit](https://github.com/tenstorrent/tt-metal/actions/workflows/blackhole-post-commit.yaml) CI passes (if applicable) - [ ] [Model regression](https://github.com/tenstorrent/tt-metal/actions/workflows/perf-models.yaml) CI passes (if applicable) - [ ] [Device performance regression](https://github.com/tenstorrent/tt-metal/actions/workflows/perf-device-models.yaml) CI passes (if applicable) - [ ] **(For models and ops writers)** Full [new models tests](https://github.com/tenstorrent/tt-metal/actions/workflows/full-new-models-suite.yaml) CI passes (if applicable) - [ ] New/Existing tests provide coverage for changes
- Loading branch information