From a893e9621be3812c51a4626a437b01210eecde7d Mon Sep 17 00:00:00 2001 From: Masaki Kozuki Date: Wed, 17 Jul 2024 12:07:22 +0000 Subject: [PATCH] a bit more strict check Signed-off-by: Masaki Kozuki --- thunder/tests/distributed/test_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/thunder/tests/distributed/test_ops.py b/thunder/tests/distributed/test_ops.py index 86e28e3323..cf5784b0b5 100644 --- a/thunder/tests/distributed/test_ops.py +++ b/thunder/tests/distributed/test_ops.py @@ -324,7 +324,7 @@ def foo( if bsym.sym.id in {all_gather_prim_impl.id, reduce_scatter_prim_impl.id}: comm_idx = idx if bsym.sym.id == wait_prim_impl.id: - self.assertGreater(idx, comm_idx) + self.assertGreater(idx, comm_idx + 2) common_utils.instantiate_parametrized_tests(DistributedCollectiveOpTest)