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 удалений
+5
Просмотреть файл
@@ -622,6 +622,11 @@ __device__ int rocshmem_test(T *ivars, int cmp, T val) {
return ctx_internal->test(ivars, cmp, val);
}
__global__ ATTR_NO_INLINE void rocshmem_barrier_all_kernel(){
rocshmem_barrier_all();
}
__device__ void rocshmem_barrier_all() {
GPU_DPRINTF("Function: rocshmem_barrier_all (ctx=%zd)\n",
get_internal_ctx(ROCSHMEM_CTX_DEFAULT));