ozaki_matmul_node.h File Reference

#include <string>
#include <vector>
#include "sdfg/analysis/analysis.h"
#include "sdfg/builder/structured_sdfg_builder.h"
#include "sdfg/data_flow/library_node.h"
#include "sdfg/data_flow/library_nodes/math/tensor/tensor_layout.h"
#include "sdfg/passes/expansion/lib_node_expander.h"
#include "sdfg/serializer/json_serializer.h"

Go to the source code of this file.

Classes
class	sdfg::fp_math::OzakiMatMulNode
	Y = A @ B with B given as its Ozaki slices, prepared once for many products, e.g. More...
class	sdfg::fp_math::OzakiMatMulNodeSerializer
class	sdfg::fp_math::OzakiMatMulExpander

Namespaces
namespace	sdfg
namespace	sdfg::fp_math

Functions
data_flow::LibraryNodeCode	sdfg::fp_math::LibraryNodeType_OzakiMatMul ("fp::OzakiMatMul")
bool	sdfg::fp_math::expand_ozaki_matmuls (builder::StructuredSDFGBuilder &builder, analysis::AnalysisManager &analysis_manager)
	Expands every OzakiMatMul node of the SDFG into the slicing of A, the Int8 MatMuls and the rescaling; whether any.