modified team based to_all -> reduce

Este cometimento está contido em:
Yiltan Hassan Temucin criou e Yiltan Temucin cometeu 2024-11-06 09:46:43 -06:00
1 ascendente 75ece02048
cometimento 997eb69b5a
14 ficheiros modificados
+52 -50

Não existem ficheiros que correspondam à sua pesquisa

+1 -2
Ver ficheiro
@@ -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,
+3 -3
Ver ficheiro
@@ -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>