![]() |
GHOST
1.1.2
General, Hybrid, and Optimized Sparse Toolkit
|
#include "ghost/config.h"#include "ghost/types.h"#include "ghost/densemat.h"#include "ghost/util.h"#include "ghost/tsmm_inplace.h"#include "ghost/tsmm_inplace_cu.h"#include "ghost/tsmm_inplace_plain_gen.h"#include "ghost/tsmm_inplace_varincols_plain_gen.h"#include "ghost/tsmm_inplace_varoutcols_plain_gen.h"#include "ghost/tsmm_inplace_var2_plain_gen.h"#include "ghost/math.h"#include "ghost/timing.h"#include "ghost/machine.h"#include "ghost/cpp11_fixes.h"#include "ghost/autogen.h"#include <unordered_map>#include <vector>#include "tsmm_inplace_plain.def"#include "tsmm_inplace_varincols_plain.def"#include "tsmm_inplace_varoutcols_plain.def"#include "tsmm_inplace_var2_plain.def"
Classes | |
| struct | std::hash< ghost_tsmm_inplace_parameters > |
Namespaces | |
| std | |
Functions | |
| static bool | operator== (const ghost_tsmm_inplace_parameters &a, const ghost_tsmm_inplace_parameters &b) |
| ghost_error | ghost_tsmm_inplace_valid (ghost_densemat *x, ghost_densemat *v, const char *transv, ghost_densemat *w, const char *transw, void *alpha, void *beta, int reduce, int printerror) |
| Check whether TSMM-inplace can be applied instead of GEMM with the given arguments. More... | |
| ghost_error | ghost_tsmm_inplace (ghost_densemat *x, ghost_densemat *w, void *alpha, void *beta) |
| Multiply a distributed dense tall skinny matrix with a redundant dense matrix in-place. More... | |
Variables | |
| static std::unordered_map < ghost_tsmm_inplace_parameters, ghost_tsmm_inplace_kernel > | ghost_tsmm_inplace_kernels |
| ghost_error ghost_tsmm_inplace_valid | ( | ghost_densemat * | x, |
| ghost_densemat * | v, | ||
| const char * | transv, | ||
| ghost_densemat * | w, | ||
| const char * | transw, | ||
| void * | alpha, | ||
| void * | beta, | ||
| int | reduce, | ||
| int | printerror | ||
| ) |
Check whether TSMM-inplace can be applied instead of GEMM with the given arguments.
| x | |
| v | |
| transv | |
| w | |
| transw | |
| alpha | |
| beta | |
| reduce | |
| printerror | Print an error message if application is not possible. |
|
static |
|
static |
1.8.6