diff --git a/tests/unit_tests/inference/engines/test_prefix_caching_cuda_graphs.py b/tests/unit_tests/inference/engines/test_prefix_caching_cuda_graphs.py index 52e231c1c68..a3dc42f1c71 100644 --- a/tests/unit_tests/inference/engines/test_prefix_caching_cuda_graphs.py +++ b/tests/unit_tests/inference/engines/test_prefix_caching_cuda_graphs.py @@ -263,6 +263,7 @@ def _step_and_log(): return finished, step_log + @pytest.mark.flaky_in_dev # Issue #6130 @pytest.mark.parametrize("model_type", ["transformer", "hybrid"]) @pytest.mark.parametrize("batch_structure", ["prefill", "decode", "mixed"]) @torch.inference_mode()