modified team based to_all -> reduce
Este cometimento está contido em:
1 ascendente
75ece02048
cometimento
997eb69b5a
14 ficheiros modificados
+52
-50
Não existem ficheiros que correspondam à sua pesquisa
@@ -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,
|
||||
|
||||
@@ -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>
|
||||
|
||||
Criar uma nova questão referindo esta
Bloquear um utilizador