Fix ops_sendrecv_test (#100)

microsoft · Sep 5, 2023 · d3fc54f · d3fc54f
1 parent 02ef336
commit d3fc54f
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/ark/ops/ops_sendrecv_test.cc b/ark/ops/ops_sendrecv_test.cc
@@ -32,13 +32,10 @@ void test_sendrecv_internal()
             exe.run(1);
             exe.stop();
 
-            LOG(ark::INFO, "rank ", gpu_id, " done");
-
-            // int tmp[2];
-            // ark::IpcAllGather barrier{"test_sendrecv_barrier", gpu_id, 2,
-            // tmp,
-            //                           sizeof(int)};
-            // barrier.sync();
+            int tmp[2];
+            ark::IpcAllGather barrier{"test_sendrecv_barrier", gpu_id, 2, tmp,
+                                      sizeof(int)};
+            barrier.sync();
             return ark::unittest::SUCCESS;
         });
     }