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

* add host API for enqueuing barrier on given stream
Этот коммит содержится в:
Dimple Prajapati
2025-10-15 14:29:07 -07:00
коммит произвёл GitHub
родитель 4ecdbc026c
Коммит a44b581997
10 изменённых файлов: 53 добавлений и 0 удалений
+7
Просмотреть файл
@@ -990,6 +990,13 @@ __host__ void rocshmem_barrier_all() {
get_internal_ctx(ROCSHMEM_HOST_CTX_DEFAULT)->barrier_all();
}
__host__ void rocshmem_barrier_all_on_stream(hipStream_t stream) {
DPRINTF("Host function: rocshmem_barrier_all_on_stream\n");
get_internal_ctx(ROCSHMEM_HOST_CTX_DEFAULT)->barrier_all_on_stream(stream);
}
__host__ void rocshmem_sync_all() {
DPRINTF("Host function: rocshmem_sync_all\n");