pytorch/tools/test
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
..
heuristics TD outside of test job (#118250) 2024-03-01 23:08:10 +00:00
gen_operators_yaml_test.py [gen_operators_yaml] add arguments to control include_all_overloads (#108396) 2023-09-02 17:37:36 +00:00
gen_oplist_test.py
test_cmake.py
test_codegen_model.py
test_codegen.py
test_create_alerts.py
test_executorch_custom_ops.py [torchgen] Add logic in custom ops to return empty tensor (#114143) 2023-12-08 17:03:44 +00:00
test_executorch_gen.py [executorch] Add support for method variants in aten executorch code gen (#121016) 2024-03-01 20:33:02 +00:00
test_executorch_signatures.py [BE] Enable ruff's UP rules and autoformat tools and scripts (#105428) 2023-07-19 01:24:44 +00:00
test_executorch_types.py
test_executorch_unboxing.py
test_gen_backend_stubs.py
test_selective_build.py
test_test_run.py Numbers based TD (#119901) 2024-02-26 17:01:19 +00:00
test_test_selections.py Fix round robin sharding (#121022) 2024-03-11 17:30:12 +00:00
test_upload_stats_lib.py TD outside of test job (#118250) 2024-03-01 23:08:10 +00:00
test_upload_test_stats.py [ez] Remove unused code in upload_test_stats (#111504) 2023-10-19 16:09:15 +00:00
test_utils.py
test_vulkan_codegen.py [pt-vulkan] Enable Python code blocks in shader templates and upgrade shader template generation (#115948) 2023-12-20 05:47:33 +00:00