Add host API for enqueuing barrier on given stream (#274)
* add host API for enqueuing barrier on given stream
[ROCm/rocshmem commit: a44b581997]
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
fc73e4f858
Коммит
6c4325d131
@@ -116,4 +116,10 @@ __host__ void Context::barrier_all() {
|
||||
HOST_DISPATCH(barrier_all());
|
||||
}
|
||||
|
||||
__host__ void Context::barrier_all_on_stream(hipStream_t stream) {
|
||||
ctxHostStats.incStat(NUM_HOST_BARRIER_ALL);
|
||||
|
||||
HOST_DISPATCH(barrier_all_on_stream(stream));
|
||||
}
|
||||
|
||||
} // namespace rocshmem
|
||||
|
||||
Ссылка в новой задаче
Block a user