frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use smaller shapes in lstm test to fix the CI timeout (#116453)

Browse Source 
Fixes https://github.com/pytorch/pytorch/issues/108824 by using smaller shapes while keeping the same test scope

Pull Request resolved: https://github.com/pytorch/pytorch/pull/116453
Approved by: https://github.com/huydhn, https://github.com/jgong5
This commit is contained in:
chunyuan
2023-12-27 09:07:05 +00:00
committed by PyTorch MergeBot
parent 499ca71e49
commit 99ef47098d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/inductor/test_cpu_repro.py
View File

@ -449,14 +449,14 @@ class CPUReproTests(TestCase):
params_dict = {
"unbatched": [True, False],
"input_size": [1, 2],
"hidden_size": [5, 32],
"num_layers": [1, 3],
"hidden_size": [2],
"num_layers": [1, 2],
"bidirectional": [False, True],
"bias": [False, True],
"empty_state": [False, True],
"batch_first": [True, False],
"batch_size": [1, 2],
"seq_len": [1, 3],
"seq_len": [1, 2],
}
self._test_lstm_packed(params_dict)