#include <mpi.h>
#include <string>
#include "common/distutils.h"
#include "common/utils.h"
#include "common/utils.hpp"
Go to the source code of this file.
|
void | mpitic () |
|
void | mpitic (int rank) |
|
double | mpitoc (int rank) |
|
double | mpitoc () |
|
void | memusage (const int myrank, std::string event) |
| Captures the memory usage of every mpi process. More...
|
|
int | itersplit (int n, int p, int r) |
| The dimension a particular rank holds out of the global dimension n across p processes. More...
|
|
int | startidx (int n, int p, int r) |
| Returns the start idx of the current rank r for a global dimension n across p processes. More...
|
|
◆ itersplit()
int itersplit |
( |
int |
n, |
|
|
int |
p, |
|
|
int |
r |
|
) |
| |
|
inline |
The dimension a particular rank holds out of the global dimension n across p processes.
- Parameters
-
[in] | n | is the global size of the tensor on that dimension |
[in] | p | is the number of splits of n. Typically number of processes on a particular mode |
[in] | r | is the rank of the mpi process in that mode. fiber rank |
Definition at line 78 of file distutils.hpp.
◆ memusage()
void memusage |
( |
const int |
myrank, |
|
|
std::string |
event |
|
) |
| |
|
inline |
Captures the memory usage of every mpi process.
Definition at line 49 of file distutils.hpp.
◆ mpitic() [1/2]
◆ mpitic() [2/2]
◆ mpitoc() [1/2]
double mpitoc |
( |
int |
rank | ) |
|
|
inline |
◆ mpitoc() [2/2]
◆ startidx()
int startidx |
( |
int |
n, |
|
|
int |
p, |
|
|
int |
r |
|
) |
| |
|
inline |
Returns the start idx of the current rank r for a global dimension n across p processes.
- Parameters
-
[in] | n | is the size of the tensor on that dimension |
[in] | p | is the number of splits of n. |
[in] | r | is the rank of the mpi process in that mode. fiber rank |
Definition at line 90 of file distutils.hpp.