Cleanup/remove a2a buff (#232)

* Do not allocate a large unused buffer for alltoall

* Remove ATA_MAX_WRKDATA_SIZE from include/rocshmem_common.hpp

[ROCm/rocshmem commit: 30ba79bcad]
This commit is contained in:
Aurelien Bouteiller
2025-09-08 10:55:06 -04:00
committato da GitHub
parent 70294d8e8c
commit fb7d89c7e6
6 ha cambiato i file con 2 aggiunte e 19 eliminazioni
+1 -7
Vedi File
@@ -399,8 +399,7 @@ void GDABackend::setup_wrk_sync_buffer() {
* Accommodate largest possible data type for pWrk
*/
wrk_sync_pool_size_ += sizeof(double) * max_num_teams *
(ROCSHMEM_REDUCE_MIN_WRKDATA_SIZE +
ROCSHMEM_ATA_MAX_WRKDATA_SIZE);
ROCSHMEM_REDUCE_MIN_WRKDATA_SIZE;
/**
* Size of fence array
@@ -479,11 +478,6 @@ void GDABackend::setup_teams() {
wrk_sync_pool_top_ += sizeof(double) * ROCSHMEM_REDUCE_MIN_WRKDATA_SIZE
* max_num_teams;
pAta_pool = reinterpret_cast<void *>(wrk_sync_pool_top_);
wrk_sync_pool_top_ += sizeof(double) * ROCSHMEM_ATA_MAX_WRKDATA_SIZE
* max_num_teams;
/**
* Initialize the sync arrays in the pool with default values.
*/