diff --git a/tests/e2e/nightly/single_node/ops/singlecard_ops/test_copy_and_expand_eagle_inputs.py b/tests/e2e/nightly/single_node/ops/singlecard_ops/test_copy_and_expand_eagle_inputs.py index 4e65b876..0a3cc3d9 100644 --- a/tests/e2e/nightly/single_node/ops/singlecard_ops/test_copy_and_expand_eagle_inputs.py +++ b/tests/e2e/nightly/single_node/ops/singlecard_ops/test_copy_and_expand_eagle_inputs.py @@ -237,7 +237,9 @@ def generate_test_case(rng, num_reqs, num_padding_slots, shift_input_ids, # --------------------------------------------------------------------------- # Parametrized tests # --------------------------------------------------------------------------- - +@pytest.mark.skip( + reason="Failure of an individual operator use case causes failures of other operators." +) @pytest.mark.parametrize("num_reqs", [1, 2, 4, 8, 16]) @pytest.mark.parametrize("num_padding_slots", [1, 2, 3, 5]) @pytest.mark.parametrize("shift_input_ids", [False, True]) @@ -301,7 +303,9 @@ def test_copy_and_expand_eagle_inputs(num_reqs, num_padding_slots, torch.testing.assert_close(n_hsm, g_hsm_t, atol=0, rtol=0, msg="out_hidden_state_mapping mismatch") - +@pytest.mark.skip( + reason="Failure of an individual operator use case causes failures of other operators." +) @pytest.mark.parametrize("num_reqs", [1]) @pytest.mark.parametrize("num_padding_slots", [1]) @pytest.mark.parametrize("shift_input_ids", [False, True]) @@ -385,7 +389,9 @@ def test_large_tokens_per_request(num_reqs): torch.testing.assert_close(n_msk, torch.from_numpy(g_msk), atol=0, rtol=0) torch.testing.assert_close(n_nti, torch.from_numpy(g_nti), atol=0, rtol=0) - +@pytest.mark.skip( + reason="Failure of an individual operator use case causes failures of other operators." +) @pytest.mark.parametrize("num_reqs", [3, 7, 13]) def test_large_tokens_shift_true(num_reqs): """Test with larger token counts and shift_input_ids=True.""" @@ -428,7 +434,9 @@ def test_large_tokens_shift_true(num_reqs): torch.testing.assert_close(n_nti, torch.from_numpy(g_nti), atol=0, rtol=0) torch.testing.assert_close(n_hsm, torch.from_numpy(g_hsm), atol=0, rtol=0) - +@pytest.mark.skip( + reason="Failure of an individual operator use case causes failures of other operators." +) @pytest.mark.parametrize("num_reqs", [1, 4, 8]) def test_no_rejected_tokens(num_reqs): """Test cases with zero rejected tokens."""