From 3f2ade22cb043ded33d7fde801082d1f37fc5aef Mon Sep 17 00:00:00 2001
From: Ubuntu <mahdiehghazi@microsoft.com>
Date: Tue, 7 Apr 2026 01:40:15 +0000
Subject: [PATCH] add barrier

---
 python/test/executor_test.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/test/executor_test.py b/python/test/executor_test.py
index eeace1a1..1175d629 100644
--- a/python/test/executor_test.py
+++ b/python/test/executor_test.py
@@ -230,6 +230,8 @@ def main(
     # Count how many elements changed
     changed = cp.count_nonzero(result_buf != cp.float16(123.0)).item()
     print("changed elements:", changed, "out of", result_buf.size)
+    cp.cuda.runtime.deviceSynchronize()
+    mscclpp_group.barrier()
 
     bench_correctness(
         collective,