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... | |
void | set_nnz_blk_A (int64_t const *nnbA) |
tspsum * | clone () |
tspsum_virt (tspsum *other) | |
iterates over the dense virtualization block grid and contracts More... | |
~tspsum_virt () | |
tspsum_virt (summation const *s) | |
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 |
int | num_dim |
int * | virt_dim |
int | order_A |
int64_t | blk_sz_A |
int const * | idx_map_A |
int | order_B |
int64_t | blk_sz_B |
int const * | idx_map_B |
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 30 of file spsum_tsr.h.
CTF_int::tspsum_virt::tspsum_virt | ( | tspsum * | other | ) |
iterates over the dense virtualization block grid and contracts
Definition at line 59 of file spsum_tsr.cxx.
References CTF_int::alloc(), blk_sz_A, blk_sz_B, CTF_int::tspsum::clone(), idx_map_A, idx_map_B, CTF_int::tspsum::nnz_blk_A, CTF_int::tspsum::nnz_blk_B, num_dim, order_A, order_B, rec_tsum, and virt_dim.
Referenced by clone().
CTF_int::tspsum_virt::~tspsum_virt | ( | ) |
Definition at line 54 of file spsum_tsr.cxx.
References CTF_int::cdealloc().
CTF_int::tspsum_virt::tspsum_virt | ( | summation const * | s | ) |
Definition at line 77 of file spsum_tsr.cxx.
References CTF_int::summation::A, CTF_int::summation::B, CTF_int::summation::idx_A, CTF_int::summation::idx_B, idx_map_A, idx_map_B, CTF_int::tensor::order, order_A, and order_B.
|
virtual |
Reimplemented from CTF_int::tspsum.
Definition at line 84 of file spsum_tsr.cxx.
References tspsum_virt().
|
virtual |
returns the number of bytes of buffer space needed
Reimplemented from CTF_int::tsum.
Definition at line 99 of file spsum_tsr.cxx.
References num_dim, order_A, and order_B.
Referenced by run().
|
virtual |
Reimplemented from CTF_int::tsum.
Definition at line 88 of file spsum_tsr.cxx.
References blk_sz_A, blk_sz_B, num_dim, CTF_int::tsum::print(), rec_tsum, and virt_dim.
|
virtual |
Reimplemented from CTF_int::tsum.
Definition at line 103 of file spsum_tsr.cxx.
References CTF_int::tsum::A, CTF_int::alloc(), CTF_int::alloc_ptr(), CTF_int::tsum::alpha, ASSERT, CTF_int::tsum::B, CTF_int::tsum::beta, blk_sz_A, blk_sz_B, CTF_int::tsum::buffer, CTF_int::cdealloc(), CTF_int::algstrct::copy_pairs(), CTF_int::accumulatable::el_size, CTF_int::tspsum::is_sparse_A, CTF_int::tspsum::is_sparse_B, mem_fp(), CTF_int::algstrct::mulid(), 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, num_dim, order_A, order_B, CTF_int::algstrct::pair_alloc(), CTF_int::algstrct::pair_dealloc(), CTF_int::algstrct::pair_size(), rec_tsum, CTF_int::tsum::run(), SET_LDA_X, CTF_int::tsum::sr_A, CTF_int::tsum::sr_B, TAU_FSTART, TAU_FSTOP, and virt_dim.
|
inlinevirtual |
Reimplemented from CTF_int::tspsum.
Definition at line 47 of file spsum_tsr.h.
References CTF_int::tspsum::clone(), and CTF_int::tspsum::set_nnz_blk_A().
int64_t CTF_int::tspsum_virt::blk_sz_A |
Definition at line 38 of file spsum_tsr.h.
Referenced by CTF_int::summation::estimate_time(), print(), run(), and tspsum_virt().
int64_t CTF_int::tspsum_virt::blk_sz_B |
Definition at line 41 of file spsum_tsr.h.
Referenced by CTF_int::summation::estimate_time(), print(), run(), and tspsum_virt().
int const* CTF_int::tspsum_virt::idx_map_A |
Definition at line 39 of file spsum_tsr.h.
Referenced by tspsum_virt().
int const* CTF_int::tspsum_virt::idx_map_B |
Definition at line 42 of file spsum_tsr.h.
Referenced by tspsum_virt().
int CTF_int::tspsum_virt::num_dim |
Definition at line 35 of file spsum_tsr.h.
Referenced by CTF_int::summation::estimate_time(), mem_fp(), print(), run(), and tspsum_virt().
int CTF_int::tspsum_virt::order_A |
Definition at line 37 of file spsum_tsr.h.
Referenced by mem_fp(), run(), and tspsum_virt().
int CTF_int::tspsum_virt::order_B |
Definition at line 40 of file spsum_tsr.h.
Referenced by mem_fp(), run(), and tspsum_virt().
tspsum* CTF_int::tspsum_virt::rec_tsum |
Definition at line 33 of file spsum_tsr.h.
Referenced by CTF_int::summation::estimate_time(), print(), CTF_int::tspsum_replicate::print(), run(), tspsum_virt(), and CTF_int::tspsum_replicate::~tspsum_replicate().
int* CTF_int::tspsum_virt::virt_dim |
Definition at line 36 of file spsum_tsr.h.
Referenced by CTF_int::summation::estimate_time(), print(), run(), and tspsum_virt().