Skip to content

Commit 7b19aa4

Browse files
committed
More shards
Signed-off-by: Dmitry Rogozhkin <[email protected]>
1 parent 1f71411 commit 7b19aa4

File tree

1 file changed

+16
-4
lines changed

1 file changed

+16
-4
lines changed

.github/workflows/_linux_transformers.yml

Lines changed: 16 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -214,16 +214,28 @@ jobs:
214214
# * https://github.com/pytorch/pytorch/issues/140965 (aten::_linalg_eigvals)
215215
# * https://github.com/huggingface/transformers/issues/36267 (marian tests)
216216
- test_case: 'tests_models_0'
217-
cmd: 'tests/models --num-shards 4 --shard-id 0 --ignore=tests/models/marian/test_modeling_marian.py'
217+
cmd: 'tests/models --num-shards 8 --shard-id 0 --ignore=tests/models/marian/test_modeling_marian.py'
218218
filter: 'not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
219219
- test_case: 'tests_models_1'
220-
cmd: 'tests/models --num-shards 4 --shard-id 1 --ignore=tests/models/marian/test_modeling_marian.py'
220+
cmd: 'tests/models --num-shards 8 --shard-id 1 --ignore=tests/models/marian/test_modeling_marian.py'
221221
filter: 'not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
222222
- test_case: 'tests_models_2'
223-
cmd: 'tests/models --num-shards 4 --shard-id 2 --ignore=tests/models/marian/test_modeling_marian.py'
223+
cmd: 'tests/models --num-shards 8 --shard-id 2 --ignore=tests/models/marian/test_modeling_marian.py'
224224
filter: 'not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
225225
- test_case: 'tests_models_3'
226-
cmd: 'tests/models --num-shards 4 --shard-id 3 --ignore=tests/models/marian/test_modeling_marian.py'
226+
cmd: 'tests/models --num-shards 8 --shard-id 3 --ignore=tests/models/marian/test_modeling_marian.py'
227+
filter: 'not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
228+
- test_case: 'tests_models_4'
229+
cmd: 'tests/models --num-shards 8 --shard-id 4 --ignore=tests/models/marian/test_modeling_marian.py'
230+
filter: 'not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
231+
- test_case: 'tests_models_5'
232+
cmd: 'tests/models --num-shards 8 --shard-id 5 --ignore=tests/models/marian/test_modeling_marian.py'
233+
filter: 'not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
234+
- test_case: 'tests_models_6'
235+
cmd: 'tests/models --num-shards 8 --shard-id 6 --ignore=tests/models/marian/test_modeling_marian.py'
236+
filter: 'not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
237+
- test_case: 'tests_models_7'
238+
cmd: 'tests/models --num-shards 8 --shard-id 7 --ignore=tests/models/marian/test_modeling_marian.py'
227239
filter: 'not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
228240
# Excluding tests due to:
229241
# * Some ray tests hang, reason unknown

0 commit comments

Comments
 (0)