fix barrier synchronization on gfx90a

Этот коммит содержится в:
Edgar Gabriel
2024-10-22 17:22:51 +00:00
родитель a0ac7b2d60
Коммит 87db7f7d38
2 изменённых файлов: 4 добавлений и 0 удалений
+3
Просмотреть файл
@@ -47,6 +47,9 @@ __device__ void IPCContext::internal_direct_barrier(int pe, int PE_start,
// Announce to other PEs that all have reached
for (size_t i = 1, j = PE_start + stride; i < n_pes; ++i, j += stride) {
internal_putmem(&pSync[0], &flag_val, sizeof(*pSync), j);
#if defined(__gfx90a__)
__threadfence_system();
#endif /* __gfx90a__ */
}
} else {
// Mark current PE offset as reached