pytorch/tools/testing
Catherine Lee 6801595349 Fix round robin sharding (#121022)
Fix round robin sharding when there are no test times and sort_by_time=False

Adds more tests to test_test_selections for sort_by_time=False
Adds more checks to test_split_shards_random for serial/parallel ordering + ordering of tests
Refactoring of dup code

Tested locally by running `python test/run_test.py --shard 3 5` with no test times downloaded and checked that it wasn't an empty list.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/121022
Approved by: https://github.com/huydhn, https://github.com/osalpekar
2024-03-11 17:30:12 +00:00
..
target_determination TD Heuristic for tests mentioned in PR body, less verbose TD printing (#120621) 2024-03-07 19:36:11 +00:00
__init__.py
discover_tests.py [ez] Discover tests without importing torch (#118574) 2024-01-30 03:02:29 +00:00
do_target_determination_for_s3.py TD outside of test job (#118250) 2024-03-01 23:08:10 +00:00
explicit_ci_jobs.py [BE] Enable ruff's UP rules and autoformat tools and scripts (#105428) 2023-07-19 01:24:44 +00:00
modulefinder_determinator.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
test_run.py Fix round robin sharding (#121022) 2024-03-11 17:30:12 +00:00
test_selections.py Fix round robin sharding (#121022) 2024-03-11 17:30:12 +00:00