Skip to content

Commit 156dcb2

Browse files
authored
Run Evoformer tests sequentially (deepspeedai#7810)
Evoformer tests fail when we run them in parallel with other tests. ``` RuntimeError: Cannot re-initialize CUDA in forked subprocess. ``` This PR adds `@pytest.mark.sequential` to the tests. See the full test log for details: https://github.com/deepspeedai/DeepSpeed/actions/runs/21303530770/job/61326548592 Signed-off-by: Masahiro Tanaka <mtanaka@anyscale.com>
1 parent 5b2ccad commit 156dcb2

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/unit/ops/deepspeed4science/test_DS4Sci_EvoformerAttention.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ def attention_reference(
4040
return o
4141

4242

43+
@pytest.mark.sequential
4344
@pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16])
4445
@pytest.mark.parametrize("tensor_shape", [(1, 256, 256, 4, 32), (1, 512, 256, 8, 8)])
4546
def test_DS4Sci_EvoformerAttention(dtype, tensor_shape):

0 commit comments

Comments
 (0)