#include <kernel.h>

Inheritance diagram for CTF::Bivar_Kernel< dtype_A, dtype_B, dtype_C, f, g >:

[legend]

Collaboration diagram for CTF::Bivar_Kernel< dtype_A, dtype_B, dtype_C, f, g >:

[legend]

Public Member Functions
	Bivar_Kernel ()

	Bivar_Kernel (bool is_comm)

void	ccoomm (int m, int n, int k, char const A, int const rows_A, int const cols_A, int64_t nnz_A, char const B, char *C) const

void	cgemm (char tA, char tB, int m, int n, int k, char const A, char const B, char *C) const

void	csrmultcsr_old (int m, int n, int k, dtype_A const A, int const JA, int const IA, int nnz_A, dtype_B const B, int const JB, int const IB, int nnz_B, char *&C_CSR) const

void	csrmultcsr (int m, int n, int k, dtype_A const A, int const JA, int const IA, int nnz_A, dtype_B const B, int const JB, int const IB, int nnz_B, char *&C_CSR) const

void	ccsrmultd (int m, int n, int k, char const A, int const JA, int const IA, int nnz_A, char const B, int const JB, int const IB, int nnz_B, char C, CTF_int::algstrct const sr_C) const

void	ccsrmultcsr (int m, int n, int k, char const A, int const JA, int const IA, int nnz_A, char const B, int const JB, int const IB, int nnz_B, char &C_CSR, CTF_int::algstrct const sr_C) const

void	ccsrmm (int m, int n, int k, char const A, int const JA, int const IA, int64_t nnz_A, char const B, char C, CTF_int::algstrct const sr_C) const

void	coffload_gemm (char tA, char tB, int m, int n, int k, char const A, char const B, char *C) const

void	coffload_csrmm (int m, int n, int k, char const all_data, char const B, char *C) const

Public Member Functions inherited from CTF::Monoid_Kernel< dtype_C, g >
	Monoid_Kernel ()

void	accum (char const a, char b) const
	b+=a More...

virtual void	init_shell (int64_t n, char *arr) const
	initialize n objects to zero More...

Public Member Functions inherited from CTF::Bivar_Function< dtype_A, dtype_B, dtype_C >
	Bivar_Function (std::function< dtype_C(dtype_A, dtype_B)> f_)
	constructor takes function pointers to compute C=f(A,B); More...

	Bivar_Function (std::function< dtype_C(dtype_A, dtype_B)> f_, bool is_comm)
	constructor takes function pointers to compute C=f(A,B); More...

	Bivar_Function ()
	default constructor sets function pointer to NULL More...

void	apply_f (char const a, char const b, char *c) const
	compute c = f(a,b) More...

void	acc_f (char const a, char const b, char c, CTF_int::algstrct const sr_C) const
	compute c = c+ f(a,b) More...

void	csrmm (int m, int n, int k, dtype_A const A, int const JA, int const IA, int64_t nnz_A, dtype_B const B, dtype_C C, CTF_int::algstrct const sr_C) const

void	csrmultd (int m, int n, int k, dtype_A const A, int const JA, int const IA, int nnz_A, dtype_B const B, int const JB, int const IB, int nnz_B, dtype_C C, CTF_int::algstrct const sr_C) const

void	csrmultcsr (int m, int n, int k, dtype_A const A, int const JA, int const IA, int nnz_A, dtype_B const B, int const JB, int const IB, int nnz_B, char &C_CSR, CTF_int::algstrct const sr_C) const

Public Member Functions inherited from CTF_int::bivar_function
void	operator() (Term const &A, Term const &B, Term const &C) const
	evaluate C+=f(A,B) or f(A,B,C) if transform More...

Bifun_Term	operator() (Term const &A, Term const &B) const
	evaluate f(A,B) More...

	bivar_function (bool is_comm=false, bool is_left_dist=false, bool is_right_dist=false)
	constructor sets function properties, pessimistic defaults More...

virtual	~bivar_function ()

virtual bool	is_accumulator () const

Static Public Member Functions
static void	gemm (char tA, char tB, int m, int n, int k, dtype_A const A, dtype_B const B, dtype_C *C)

static void	coomm (int m, int n, int k, dtype_A const A, int const rows_A, int const cols_A, int nnz_A, dtype_B const B, dtype_C *C)

static void	csrmm (int m, int n, int k, dtype_A const A, int const JA, int const IA, int64_t nnz_A, dtype_B const B, dtype_C *C)

static void	csrmultd (int m, int n, int k, dtype_A const A, int const JA, int const IA, int nnz_A, dtype_B const B, int const JB, int const IB, int nnz_B, dtype_C *C)

static void	offload_gemm (char tA, char tB, int m, int n, int k, dtype_A const A, dtype_B const B, dtype_C *C)

Static Public Member Functions inherited from CTF::Monoid_Kernel< dtype_C, g >
static MPI_Op	get_MPI_Op ()

static void	xpy (int n, dtype_C const X, int incX, dtype_C Y, int incY)

Additional Inherited Members
Data Fields inherited from CTF_int::accumulatable
int	el_size
	size of each element of algstrct in bytes More...

Data Fields inherited from CTF::Bivar_Function< dtype_A, dtype_B, dtype_C >
std::function< dtype_C(dtype_A, dtype_B)>	f
	function signature for element-wise multiplication, compute C=f(A,B) More...

Data Fields inherited from CTF_int::bivar_function
bool	has_kernel

bool	has_off_gemm

bool	commutative

bool	left_distributive

bool	right_distributive