fix barrier synchronization on gfx90a

Esse commit está contido em:
Edgar Gabriel
2024-10-22 17:22:51 +00:00
commit 87db7f7d38
2 arquivos alterados com 4 adições e 0 exclusões
+1
Ver Arquivo
@@ -58,6 +58,7 @@ __host__ IPCContext::IPCContext(Backend *b)
}
__device__ void IPCContext::threadfence_system() {
__threadfence_system();
}
__device__ void IPCContext::ctx_create() {
+3
Ver Arquivo
@@ -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