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 Revert "Rename DispatchKey.PrivateUse1 to custom device in torchgen. (#99406)" 2023-05-09 15:04:16 +00:00
test_codegen.py [torchgen] Let native function declaration generation logic take a callable (#90780) 2022-12-14 20:13:04 +00:00
test_create_alerts.py Adds script to generate alerts for failing jobs (#102002) 2023-05-31 23:20:31 +00:00
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 [torchgen] Introduce Executorch types and signatures (#90781) 2022-12-14 20:13:04 +00:00
test_executorch_unboxing.py [torchgen] Move Executorch unboxing logic into torchgen (#90098) 2022-12-19 21:58:43 +00:00
test_gen_backend_stubs.py Bump black version to 23.1.0 (#96578) 2023-03-15 06:27:59 +00:00
test_selective_build.py [Reland][ET] Select used et_kernel_metadata only (#104005) 2023-06-23 14:38:45 +00:00
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