lightning-thunder
lightning-thunder copied to clipboard
Reduces the test time
Before submitting
- [ ] Was this discussed/approved via a Github issue? (no need for typos and docs improvements)
- [ ] Did you read the contributor guideline, Pull Request section?
- [ ] Did you make sure to update the docs?
- [ ] Did you write any new necessary tests?
What does this PR do?
Fixes #2051 .
43.38s call thunder/tests/test_dynamo.py::test_report_thunderfx_pytest_benchmark_report
42.76s call thunder/tests/test_dynamo.py::test_fxreport_DynamoThunder_cuda_None[benchmark]
38.60s call thunder/tests/test_dynamo.py::test_ThunderCompilerGraphBenchmarking_groupby
18.62s call thunder/tests/test_dynamo.py::test_reports_repro
17.10s call thunder/tests/test_dynamo.py::test_thunder_specific_reports
19.96s call thunder/tests/test_dynamo.py::test_reports_benchmark
https://dev.azure.com/Lightning-AI/lightning/_build/results?buildId=232923&view=logs&jobId=5b0799f7-725e-5b16-9b83-c0a5a25d03f0&j=3f274fac-2e11-54ca-487e-194c91f3ae9f&t=7d03ab59-8b52-555c-a216-77e14d28deec
Hi @mruberry @t-vi , I modified to use hypothesis and the test time is updated in the PR description