sdfg::einsum::Einsum2MatMul::MatMulAnalysis Struct Reference

Results of the applicability analysis that are reused by the replacement step. More...

#include <sdfg/einsum/replacers/einsum2matmul.h>

Inheritance diagram for sdfg::einsum::Einsum2MatMul::MatMulAnalysis:

Public Attributes
symbolic::Symbol	indvar_outer_1 = SymEngine::null
symbolic::Symbol	indvar_outer_2 = SymEngine::null
symbolic::Symbol	indvar_inner = SymEngine::null
symbolic::Expression	m = SymEngine::null
symbolic::Expression	n = SymEngine::null
symbolic::Expression	k = SymEngine::null
symbolic::Expression	m_init = SymEngine::null
symbolic::Expression	n_init = SymEngine::null
symbolic::Expression	k_init = SymEngine::null
int	a_idx = -1
int	b_idx = -1
std::optional< math::tensor::TensorLayout >	layout_a
std::optional< math::tensor::TensorLayout >	layout_b
std::optional< math::tensor::TensorLayout >	layout_y
types::PrimitiveType	input_type = types::PrimitiveType::Void
	whether the indices in our contributors array are swapped to how we canonically name them this is not "transposed" as that will simply depend on the strides associated with the respective indices.
types::PrimitiveType	output_type = types::PrimitiveType::Void

Detailed Description

Results of the applicability analysis that are reused by the replacement step.

Member Data Documentation

a_idx

int sdfg::einsum::Einsum2MatMul::MatMulAnalysis::a_idx = -1

b_idx

int sdfg::einsum::Einsum2MatMul::MatMulAnalysis::b_idx = -1

indvar_inner

symbolic::Symbol sdfg::einsum::Einsum2MatMul::MatMulAnalysis::indvar_inner = SymEngine::null

indvar_outer_1

symbolic::Symbol sdfg::einsum::Einsum2MatMul::MatMulAnalysis::indvar_outer_1 = SymEngine::null

indvar_outer_2

symbolic::Symbol sdfg::einsum::Einsum2MatMul::MatMulAnalysis::indvar_outer_2 = SymEngine::null

input_type

types::PrimitiveType sdfg::einsum::Einsum2MatMul::MatMulAnalysis::input_type = types::PrimitiveType::Void

whether the indices in our contributors array are swapped to how we canonically name them this is not "transposed" as that will simply depend on the strides associated with the respective indices.

This is just so we use the stride matching the indvars

k

symbolic::Expression sdfg::einsum::Einsum2MatMul::MatMulAnalysis::k = SymEngine::null

k_init

symbolic::Expression sdfg::einsum::Einsum2MatMul::MatMulAnalysis::k_init = SymEngine::null

layout_a

std::optional<math::tensor::TensorLayout> sdfg::einsum::Einsum2MatMul::MatMulAnalysis::layout_a

layout_b

std::optional<math::tensor::TensorLayout> sdfg::einsum::Einsum2MatMul::MatMulAnalysis::layout_b

layout_y

std::optional<math::tensor::TensorLayout> sdfg::einsum::Einsum2MatMul::MatMulAnalysis::layout_y

m

symbolic::Expression sdfg::einsum::Einsum2MatMul::MatMulAnalysis::m = SymEngine::null

m_init

symbolic::Expression sdfg::einsum::Einsum2MatMul::MatMulAnalysis::m_init = SymEngine::null

n

symbolic::Expression sdfg::einsum::Einsum2MatMul::MatMulAnalysis::n = SymEngine::null

n_init

symbolic::Expression sdfg::einsum::Einsum2MatMul::MatMulAnalysis::n_init = SymEngine::null

output_type

types::PrimitiveType sdfg::einsum::Einsum2MatMul::MatMulAnalysis::output_type = types::PrimitiveType::Void

---

The documentation for this struct was generated from the following file:

• opt/include/sdfg/einsum/replacers/einsum2matmul.h