File tree Expand file tree Collapse file tree 2 files changed +6
-1
lines changed
Expand file tree Collapse file tree 2 files changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -48,7 +48,8 @@ def _get_test_seed(test_base_name: str) -> int:
4848 hasher = hashlib .md5 ()
4949 data = test_base_name .encode ("utf-8" )
5050 hasher .update (data )
51- return int .from_bytes (hasher .hexdigest (), "little" )
51+ # Torch doesn't like very long seeds.
52+ return int .from_bytes (hasher .digest (), "little" ) % 100_000_000
5253
5354def run_test ( # noqa: C901
5455 model : torch .nn .Module ,
Original file line number Diff line number Diff line change 2424 base_name = "test1" ,
2525 flow = "flow1" ,
2626 name = "test1_backend1_flow1" ,
27+ subtest_index = 0 ,
2728 params = None ,
2829 result = TestResult .SUCCESS ,
2930 error = None ,
3435 base_name = "test1" ,
3536 flow = "flow1" ,
3637 name = "test1_backend2_flow1" ,
38+ subtest_index = 0 ,
3739 params = None ,
3840 result = TestResult .LOWER_FAIL ,
3941 error = None ,
4446 base_name = "test2" ,
4547 flow = "flow1" ,
4648 name = "test2_backend1_flow1" ,
49+ subtest_index = 0 ,
4750 params = {"dtype" : torch .float32 },
4851 result = TestResult .SUCCESS_UNDELEGATED ,
4952 error = None ,
5457 base_name = "test2" ,
5558 flow = "flow1" ,
5659 name = "test2_backend2_flow1" ,
60+ subtest_index = 0 ,
5761 params = {"use_dynamic_shapes" : True },
5862 result = TestResult .SKIPPED ,
5963 error = None ,
You can’t perform that action at this time.
0 commit comments