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]
Bu işleme şunda yer alıyor:
Aurelien Bouteiller
2025-09-08 10:55:06 -04:00
işlemeyi yapan: GitHub
ebeveyn 70294d8e8c
işleme fb7d89c7e6
6 değiştirilmiş dosya ile 2 ekleme ve 19 silme
+1 -7
Dosyayı Görüntüle
@@ -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.
*/
-1
Dosyayı Görüntüle
@@ -46,7 +46,6 @@ GDATeam::GDATeam(Backend *backend, TeamInfo *team_info_parent,
alltoall_pSync = &(b->alltoall_pSync_pool[pool_index * ROCSHMEM_ALLTOALL_SYNC_SIZE]);
pWrk = reinterpret_cast<char *>(b->pWrk_pool) + ROCSHMEM_REDUCE_MIN_WRKDATA_SIZE * sizeof(double) * pool_index;
pAta = reinterpret_cast<char *>(b->pAta_pool) + ROCSHMEM_ATA_MAX_WRKDATA_SIZE * sizeof(double) * pool_index;
}
GDATeam::~GDATeam() {}