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]
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
70294d8e8c
Коммит
fb7d89c7e6
@@ -88,7 +88,6 @@ typedef struct {
|
||||
} rocshmem_team_config_t;
|
||||
|
||||
constexpr size_t ROCSHMEM_REDUCE_MIN_WRKDATA_SIZE = 1024;
|
||||
constexpr size_t ROCSHMEM_ATA_MAX_WRKDATA_SIZE = (4 * 1024 * 1024);
|
||||
constexpr size_t ROCSHMEM_BARRIER_SYNC_SIZE = 256;
|
||||
constexpr size_t ROCSHMEM_REDUCE_SYNC_SIZE = 256;
|
||||
// Internally calls sync function, which matches barrier implementation
|
||||
|
||||
Ссылка в новой задаче
Block a user