Cyclops Tensor Framework
parallel arithmetic on multidimensional arrays
|
#include <spsum_tsr.h>
Public Member Functions | |
void | run () |
void | print () |
int64_t | mem_fp () |
returns the number of bytes of buffer space needed More... | |
tspsum * | clone () |
void | set_nnz_blk_A (int64_t const *nnbA) |
tspsum_permute (tspsum *other) | |
~tspsum_permute () | |
tspsum_permute (summation const *s, bool A_or_B, int const *lens) | |
Public Member Functions inherited from CTF_int::tspsum | |
~tspsum () | |
tspsum (tspsum *other) | |
tspsum (summation const *s) | |
Public Member Functions inherited from CTF_int::tsum | |
virtual | ~tsum () |
tsum (tsum *other) | |
tsum (summation const *s) | |
Data Fields | |
tspsum * | rec_tsum |
bool | A_or_B |
int | order |
int * | lens_new |
int * | lens_old |
int * | p |
bool | skip |
Data Fields inherited from CTF_int::tspsum | |
bool | is_sparse_A |
int64_t | nnz_A |
int | nvirt_A |
int64_t * | nnz_blk_A |
bool | is_sparse_B |
int64_t | nnz_B |
int | nvirt_B |
int64_t * | nnz_blk_B |
int64_t | new_nnz_B |
char * | new_B |
Data Fields inherited from CTF_int::tsum | |
char * | A |
algstrct const * | sr_A |
char const * | alpha |
char * | B |
algstrct const * | sr_B |
char const * | beta |
void * | buffer |
Definition at line 156 of file spsum_tsr.h.
CTF_int::tspsum_permute::tspsum_permute | ( | tspsum * | other | ) |
Definition at line 664 of file spsum_tsr.cxx.
References A_or_B, CTF_int::alloc(), CTF_int::tspsum::clone(), lens_new, lens_old, order, p, rec_tsum, and skip.
Referenced by clone().
CTF_int::tspsum_permute::~tspsum_permute | ( | ) |
Definition at line 657 of file spsum_tsr.cxx.
References CTF_int::cdealloc(), and CTF_int::tspsum_map::rec_tsum.
CTF_int::tspsum_permute::tspsum_permute | ( | summation const * | s, |
bool | A_or_B, | ||
int const * | lens | ||
) |
Definition at line 678 of file spsum_tsr.cxx.
References CTF_int::summation::A, A_or_B, CTF_int::alloc(), ASSERT, CTF_int::summation::B, CTF_int::summation::idx_A, CTF_int::summation::idx_B, lens_new, lens_old, CTF_int::tensor::order, order, p, and skip.
|
virtual |
Reimplemented from CTF_int::tspsum.
Definition at line 768 of file spsum_tsr.cxx.
References tspsum_permute().
|
virtual |
returns the number of bytes of buffer space needed
Reimplemented from CTF_int::tsum.
Definition at line 783 of file spsum_tsr.cxx.
References A_or_B, CTF_int::tspsum::nnz_A, CTF_int::tspsum::nnz_B, CTF_int::algstrct::pair_size(), CTF_int::tsum::sr_A, and CTF_int::tsum::sr_B.
|
virtual |
Reimplemented from CTF_int::tsum.
Definition at line 772 of file spsum_tsr.cxx.
References A_or_B, order, p, CTF_int::tsum::print(), and rec_tsum.
|
virtual |
Reimplemented from CTF_int::tsum.
Definition at line 790 of file spsum_tsr.cxx.
References CTF_int::tsum::A, A_or_B, CTF_int::alloc(), CTF_int::alloc_ptr(), CTF_int::tsum::B, CTF_int::cdealloc(), CTF_int::algstrct::copy(), CTF_int::algstrct::copy_pairs(), lens_new, lens_old, CTF_int::tspsum::new_B, CTF_int::tspsum::new_nnz_B, CTF_int::tspsum::nnz_A, CTF_int::tspsum::nnz_B, CTF_int::tspsum::nnz_blk_A, CTF_int::tspsum::nnz_blk_B, CTF_int::tspsum::nvirt_A, CTF_int::tspsum::nvirt_B, order, p, CTF_int::algstrct::pair_dealloc(), CTF_int::algstrct::pair_size(), CTF_int::ConstPairIterator::permute(), rec_tsum, CTF_int::tsum::run(), skip, CTF_int::PairIterator::sort(), CTF_int::tsum::sr_A, CTF_int::tsum::sr_B, TAU_FSTART, and TAU_FSTOP.
|
inlinevirtual |
Reimplemented from CTF_int::tspsum.
Definition at line 170 of file spsum_tsr.h.
References CTF_int::tspsum::set_nnz_blk_A().
bool CTF_int::tspsum_permute::A_or_B |
Definition at line 159 of file spsum_tsr.h.
Referenced by mem_fp(), print(), run(), and tspsum_permute().
int* CTF_int::tspsum_permute::lens_new |
Definition at line 161 of file spsum_tsr.h.
Referenced by run(), and tspsum_permute().
int* CTF_int::tspsum_permute::lens_old |
Definition at line 162 of file spsum_tsr.h.
Referenced by run(), and tspsum_permute().
int CTF_int::tspsum_permute::order |
Definition at line 160 of file spsum_tsr.h.
Referenced by ctf.core.tensor::__get__(), ctf.core.tensor::__getitem__(), ctf.core.tensor::__setitem__(), ctf.core.tensor::i(), ctf.core.tensor::permute(), print(), run(), ctf.core.tensor::transpose(), and tspsum_permute().
int* CTF_int::tspsum_permute::p |
Definition at line 163 of file spsum_tsr.h.
Referenced by print(), run(), and tspsum_permute().
tspsum* CTF_int::tspsum_permute::rec_tsum |
Definition at line 158 of file spsum_tsr.h.
Referenced by CTF_int::summation::estimate_time(), print(), run(), tspsum_permute(), and CTF_int::tspsum_pin_keys::~tspsum_pin_keys().
bool CTF_int::tspsum_permute::skip |
Definition at line 164 of file spsum_tsr.h.
Referenced by run(), and tspsum_permute().