AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modified team based to_all -> reduce

Browse Source 
[ROCm/rocshmem commit: 997eb69b5a]
This commit is contained in:
Yiltan Hassan Temucin
2024-10-29 06:32:13 -07:00
committed by Yiltan Temucin
parent b43ef2b45b
commit 7f3dd33144
14 changed files with 52 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocshmem/src/host/host.hpp
+1 -2
View File
@@ -207,8 +207,7 @@ class HostInterface {
long* p_sync); // NOLINT(runtime/int)
template <typename T, ROC_SHMEM_OP Op>
__host__ void to_all(roc_shmem_team_t team, T* dest, const T* source,
int nreduce);
__host__ int reduce(roc_shmem_team_t team, T* dest, const T* source, int nreduce);
template <typename T>
__host__ void wait_until(T *ivars, int cmp, T val,
projects/rocshmem/src/host/host_templates.hpp
+3 -3
View File
@@ -376,9 +376,9 @@ __host__ void HostInterface::to_all(T* dest, const T* source, int nreduce,
}
template <typename T, ROC_SHMEM_OP Op>
__host__ void HostInterface::to_all(roc_shmem_team_t team, T* dest,
__host__ int HostInterface::reduce(roc_shmem_team_t team, T* dest,
const T* source, int nreduce) {
DPRINTF("Function: Team-based host_to_all\n");
DPRINTF("Function: Team-based host_reduce\n");
/*
* Get the MPI communicator of this team
@@ -388,7 +388,7 @@ __host__ void HostInterface::to_all(roc_shmem_team_t team, T* dest,
to_all_internal<T, Op>(mpi_comm, dest, source, nreduce);
return;
return ROC_SHMEM_SUCCESS;
}
template <typename T>