From 61776e69f9c74b57b3395a89699b9e3ed333f4fd Mon Sep 17 00:00:00 2001 From: Masahiro Tanaka Date: Fri, 23 Jan 2026 15:55:44 -0800 Subject: [PATCH] run evoformer tests sequentially Signed-off-by: Masahiro Tanaka --- .../unit/ops/deepspeed4science/test_DS4Sci_EvoformerAttention.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/unit/ops/deepspeed4science/test_DS4Sci_EvoformerAttention.py b/tests/unit/ops/deepspeed4science/test_DS4Sci_EvoformerAttention.py index 25624f3a6818..307f59157220 100644 --- a/tests/unit/ops/deepspeed4science/test_DS4Sci_EvoformerAttention.py +++ b/tests/unit/ops/deepspeed4science/test_DS4Sci_EvoformerAttention.py @@ -40,6 +40,7 @@ def attention_reference( return o +@pytest.mark.sequential @pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16]) @pytest.mark.parametrize("tensor_shape", [(1, 256, 256, 4, 32), (1, 512, 256, 8, 8)]) def test_DS4Sci_EvoformerAttention(dtype, tensor_shape):