Add host API for enqueuing barrier on given stream (#274)

* add host API for enqueuing barrier on given stream
2025-10-15 14:29:07 -07:00
parent 4ecdbc026c
commit a44b581997
10 changed files with 53 additions and 0 deletions
@@ -341,6 +341,13 @@ __host__ void rocshmem_quiet();
 */
 __host__ void rocshmem_barrier_all();

+/**
+ * @brief enqueues a collective barrier on given stream.
+ *
+ * @return void
+ */
+__host__ void rocshmem_barrier_all_on_stream(hipStream_t stream);
+
 /**
 * @brief registers the arrival of a PE at a barrier.
 * The caller is blocked until the synchronization is resolved.