We need to track the context of the code generation to support MLIR class Context: More...

Inheritance diagram for printers.MLIRPrinter.MLIRPrinter:

Collaboration diagram for printers.MLIRPrinter.MLIRPrinter:

Public Member Functions
	__init__ (self, spaces_per_tab=2, use_omp=False)

str	get_indent (self, int indent_level)

	print_function_definition (self, SyntaxTree.FunctionDefinition function_definition, indent_level=0)

	print_function_declaration (self, SyntaxTree.FunctionDefinition function_definition, indent_level=0)

	print_function_definition_with_timer (self, SyntaxTree.FunctionDefinition function_definition, indent_level=0)

	print_function_declaration_with_timer (self, SyntaxTree.FunctionDefinition function_definition, indent_level=0)

	print_argument (self, SyntaxTree.Argument argument, indent_level=0)

	print_name (self, SyntaxTree.Name name, indent_level=0)

	print_integer (self, SyntaxTree.Integer integer, indent_level=0)

	print_if (self, SyntaxTree.If if_statement, indent_level=0)

	print_for (self, SyntaxTree.For for_statement, indent_level=0)

	print_comment (self, SyntaxTree.Comment comment, indent_level=0)

	print_function_call (self, SyntaxTree.FunctionCall function_call, indent_level=0)

	print_max (self, SyntaxTree.Max max, indent_level=0)

	print_memory_allocation (self, SyntaxTree.MemoryAllocation memory_allocation, indent_level=0)

	print_memory_deallocation (self, SyntaxTree.MemoryDeallocation memory_deallocation, indent_level=0)

	print_assignment (self, SyntaxTree.Assignment assignment, indent_level=0)

	print_construction (self, SyntaxTree.Construction construction, indent_level=0)

	print_data_block_construction_from_existing (self, SyntaxTree.DataBlockConstructionFromExisting construction, indent_level=0)

	print_data_block_construction_from_operation (self, SyntaxTree.DataBlockConstructionFromOperation construction, indent_level=0)

	print_data_block_construction_from_function (self, SyntaxTree.DataBlockConstructionFromFunction construction, indent_level=0)

	print_data_block_construction_from_matrix_vector_product (self, SyntaxTree.DataBlockConstructionFromFunction construction, indent_level=0)

	print_data_block_construction_from_dot_product (self, SyntaxTree.DataBlockConstructionFromFunction construction, indent_level=0)

	print_diagonal_matrix (self, SyntaxTree.DiagonalMatrix matrix, indent_level=0)

	print_matrix (self, SyntaxTree.Matrix construction, indent_level=0)

	print_diagonal_matrix_construction_from_kronecker_product (self, SyntaxTree.DataBlockConstructionFromFunction construction, indent_level=0)

	print_diagonal_matrix_construction_from_value (self, SyntaxTree.DataBlockConstructionFromFunction construction, indent_level=0)

	print_matrix_construction_from_kronecker_product (self, SyntaxTree.DataBlockConstructionFromFunction construction, indent_level=0)

	print_data_block_assignment (self, SyntaxTree.DataBlockAssignment assignment, indent_level=0)

	print_boolean (self, SyntaxTree.Boolean boolean, indent_level=0)

	print_string (self, SyntaxTree.String string, indent_level=0)

	print_float (self, SyntaxTree.Float float, indent_level=0)

	print_data_block (self, SyntaxTree.DataBlock data_block, indent_level=0)

	print_subscript (self, SyntaxTree.Subscript subscript, indent_level=0)

	print_reference (self, SyntaxTree.Reference reference, indent_level=0)

	print_vector (self, SyntaxTree.Vector vector, indent_level=0)

	print_vector_index (self, SyntaxTree.VectorIndex vector_index, indent_level=0)

	print_comparison (self, SyntaxTree.Comparison comparison, indent_level=0)

	print_binary_operation (self, SyntaxTree.BinaryOperation binary_operation, indent_level=0)

	print_unary_operation (self, SyntaxTree.UnaryOperation unary_operation, indent_level=0)

	print_data_block_unary_max (self, SyntaxTree.DataBlockUnaryMax unary_max, indent_level=0)

	print_custom_type (self, SyntaxTree.TCustom type, indent_level=0)

	print_void_type (self, SyntaxTree.TVoid type, indent_level=0)

	print_integer_type (self, SyntaxTree.TInteger type, indent_level=0)

	print_long_type (self, SyntaxTree.TLong type, indent_level=0)

	print_boolean_type (self, SyntaxTree.TBoolean type, indent_level=0)

	print_float_type (self, SyntaxTree.TFloat type, indent_level=0)

	print_string_type (self, SyntaxTree.TString type, indent_level=0)

	print_data_block_type (self, SyntaxTree.TString type, indent_level=0)

	print_vector_type (self, SyntaxTree.TVector type, indent_level=0)

	print_nested_memref_type (self, SyntaxTree.TNestedMemref type, indent_level=0)

	print_simple_memref_type (self, SyntaxTree.TSimpleMemref type, indent_level=0)

	print_function_type (self, SyntaxTree.TFunctionType type, indent_level=0)

Data Fields
	use_omp

	cpp_printer

Protected Attributes
	_spaces_per_tab

Detailed Description

We need to track the context of the code generation to support MLIR class Context:

def __init__(self):
    # A stack of blocks e.g. for loops where we can store hoisted MLIR operations
    self._blocks = [ Context.Block() ]  

def push_block(self):
    self._blocks.append(Context.Block())

def pop_block(self):
    if self._blocks:
        return self._blocks.pop()
    else:
        return Context.Block()

def get_block(self):
    if not self._blocks:
        self.push_block()
    return self._blocks[-1]

def clear_block(self):
    if self._blocks:
        self._blocks[-1] = Context.Block()

# We need to store block-level MLIR that has been hoisted and ensure we don't
# repeat casting operators for indices within a block e.g. loops    
class Block:

    def __init__(self):
        # We use this to hoist MLIR operations out of array indexing, function calls etc.
        self._mlir = []            
        # We use a set to store the ids that we have already processed within the block
        self._ids = {}           

    def push_mlir(self):
        self._mlir.append([])

    def pop_mlir(self):
        if self._mlir:
            return self._mlir.pop()
        else:
            return []

    def get_mlir(self):
        if not self._mlir:
            self.push_mlir()
        return self._mlir[-1]    

    def mlir_append(self, op):
        self.get_mlir().append(op)  

    # def add_id(self, id):
    #     self._ids.update(id)

    # def remove_id(self, id):
    #     self._ids.remove(id)

    # def get_id(self, id):
    #     return self._ids.get(id)

    # def find_id(self, id):
    #     return id in self._ids

Definition at line 71 of file MLIRPrinter.py.

Constructor & Destructor Documentation

◆ init()

printers.MLIRPrinter.MLIRPrinter.__init__	(	self,
		spaces_per_tab = `2`,
		use_omp = `False`
	)

Definition at line 72 of file MLIRPrinter.py.

Member Function Documentation

◆ get_indent()

str printers.MLIRPrinter.MLIRPrinter.get_indent	(		self,
		int	indent_level
	)

Definition at line 77 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter._spaces_per_tab, and printers.MLIRPrinter.MLIRPrinter._spaces_per_tab.

Here is the caller graph for this function:

◆ print_argument()

printers.MLIRPrinter.MLIRPrinter.print_argument	(		self,
		SyntaxTree.Argument	argument,
			indent_level = `0`
	)

Definition at line 250 of file MLIRPrinter.py.

◆ print_assignment()

printers.MLIRPrinter.MLIRPrinter.print_assignment	(		self,
		SyntaxTree.Assignment	assignment,
			indent_level = `0`
	)

Definition at line 871 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_binary_operation()

printers.MLIRPrinter.MLIRPrinter.print_binary_operation	(		self,
		SyntaxTree.BinaryOperation	binary_operation,
			indent_level = `0`
	)

Definition at line 1580 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_boolean()

printers.MLIRPrinter.MLIRPrinter.print_boolean	(		self,
		SyntaxTree.Boolean	boolean,
			indent_level = `0`
	)

Definition at line 1143 of file MLIRPrinter.py.

◆ print_boolean_type()

printers.MLIRPrinter.MLIRPrinter.print_boolean_type	(		self,
		SyntaxTree.TBoolean	type,
			indent_level = `0`
	)

Definition at line 1688 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_comment()

printers.MLIRPrinter.MLIRPrinter.print_comment	(		self,
		SyntaxTree.Comment	comment,
			indent_level = `0`
	)

Definition at line 397 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_comparison()

printers.MLIRPrinter.MLIRPrinter.print_comparison	(		self,
		SyntaxTree.Comparison	comparison,
			indent_level = `0`
	)

Definition at line 1557 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_construction()

printers.MLIRPrinter.MLIRPrinter.print_construction	(		self,
		SyntaxTree.Construction	construction,
			indent_level = `0`
	)

Definition at line 1031 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_custom_type()

printers.MLIRPrinter.MLIRPrinter.print_custom_type	(		self,
		SyntaxTree.TCustom	type,
			indent_level = `0`
	)

Definition at line 1666 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_data_block()

printers.MLIRPrinter.MLIRPrinter.print_data_block	(		self,
		SyntaxTree.DataBlock	data_block,
			indent_level = `0`
	)

Definition at line 1287 of file MLIRPrinter.py.

◆ print_data_block_assignment()

printers.MLIRPrinter.MLIRPrinter.print_data_block_assignment	(		self,
		SyntaxTree.DataBlockAssignment	assignment,
			indent_level = `0`
	)

Definition at line 1140 of file MLIRPrinter.py.

◆ print_data_block_construction_from_dot_product()

printers.MLIRPrinter.MLIRPrinter.print_data_block_construction_from_dot_product	(		self,
		SyntaxTree.DataBlockConstructionFromFunction	construction,
			indent_level = `0`
	)

Definition at line 1109 of file MLIRPrinter.py.

◆ print_data_block_construction_from_existing()

printers.MLIRPrinter.MLIRPrinter.print_data_block_construction_from_existing	(		self,
		SyntaxTree.DataBlockConstructionFromExisting	construction,
			indent_level = `0`
	)

Definition at line 1054 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_data_block_construction_from_function()

printers.MLIRPrinter.MLIRPrinter.print_data_block_construction_from_function	(		self,
		SyntaxTree.DataBlockConstructionFromFunction	construction,
			indent_level = `0`
	)

Definition at line 1094 of file MLIRPrinter.py.

◆ print_data_block_construction_from_matrix_vector_product()

printers.MLIRPrinter.MLIRPrinter.print_data_block_construction_from_matrix_vector_product	(		self,
		SyntaxTree.DataBlockConstructionFromFunction	construction,
			indent_level = `0`
	)

Definition at line 1105 of file MLIRPrinter.py.

◆ print_data_block_construction_from_operation()

printers.MLIRPrinter.MLIRPrinter.print_data_block_construction_from_operation	(		self,
		SyntaxTree.DataBlockConstructionFromOperation	construction,
			indent_level = `0`
	)

Definition at line 1090 of file MLIRPrinter.py.

◆ print_data_block_type()

printers.MLIRPrinter.MLIRPrinter.print_data_block_type	(		self,
		SyntaxTree.TString	type,
			indent_level = `0`
	)

Definition at line 1697 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_data_block_unary_max()

printers.MLIRPrinter.MLIRPrinter.print_data_block_unary_max	(		self,
		SyntaxTree.DataBlockUnaryMax	unary_max,
			indent_level = `0`
	)

Definition at line 1663 of file MLIRPrinter.py.

◆ print_diagonal_matrix()

printers.MLIRPrinter.MLIRPrinter.print_diagonal_matrix	(		self,
		SyntaxTree.DiagonalMatrix	matrix,
			indent_level = `0`
	)

Definition at line 1113 of file MLIRPrinter.py.

◆ print_diagonal_matrix_construction_from_kronecker_product()

printers.MLIRPrinter.MLIRPrinter.print_diagonal_matrix_construction_from_kronecker_product	(		self,
		SyntaxTree.DataBlockConstructionFromFunction	construction,
			indent_level = `0`
	)

Definition at line 1126 of file MLIRPrinter.py.

References printers.MLIRPrinter.MLIRPrinter.cpp_printer.

◆ print_diagonal_matrix_construction_from_value()

printers.MLIRPrinter.MLIRPrinter.print_diagonal_matrix_construction_from_value	(		self,
		SyntaxTree.DataBlockConstructionFromFunction	construction,
			indent_level = `0`
	)

Definition at line 1131 of file MLIRPrinter.py.

◆ print_float()

printers.MLIRPrinter.MLIRPrinter.print_float	(		self,
		SyntaxTree.Float	float,
			indent_level = `0`
	)

Definition at line 1270 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_float_type()

printers.MLIRPrinter.MLIRPrinter.print_float_type	(		self,
		SyntaxTree.TFloat	type,
			indent_level = `0`
	)

Definition at line 1691 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_for()

printers.MLIRPrinter.MLIRPrinter.print_for	(		self,
		SyntaxTree.For	for_statement,
			indent_level = `0`
	)

Definition at line 298 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_function_call()

printers.MLIRPrinter.MLIRPrinter.print_function_call	(		self,
		SyntaxTree.FunctionCall	function_call,
			indent_level = `0`
	)

Definition at line 400 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_function_declaration()

printers.MLIRPrinter.MLIRPrinter.print_function_declaration	(		self,
		SyntaxTree.FunctionDefinition	function_definition,
			indent_level = `0`
	)

Definition at line 177 of file MLIRPrinter.py.

References printers.MLIRPrinter.MLIRPrinter.cpp_printer, PipelineBuilder.PipelineBuilder.use_omp, and printers.MLIRPrinter.MLIRPrinter.use_omp.

◆ print_function_declaration_with_timer()

printers.MLIRPrinter.MLIRPrinter.print_function_declaration_with_timer	(		self,
		SyntaxTree.FunctionDefinition	function_definition,
			indent_level = `0`
	)

Definition at line 227 of file MLIRPrinter.py.

References printers.MLIRPrinter.MLIRPrinter.cpp_printer, PipelineBuilder.PipelineBuilder.use_omp, and printers.MLIRPrinter.MLIRPrinter.use_omp.

◆ print_function_definition()

printers.MLIRPrinter.MLIRPrinter.print_function_definition	(		self,
		SyntaxTree.FunctionDefinition	function_definition,
			indent_level = `0`
	)

Definition at line 80 of file MLIRPrinter.py.

References printers.MLIRPrinter.MLIRPrinter.cpp_printer, PipelineBuilder.PipelineBuilder.use_omp, and printers.MLIRPrinter.MLIRPrinter.use_omp.

◆ print_function_definition_with_timer()

printers.MLIRPrinter.MLIRPrinter.print_function_definition_with_timer	(		self,
		SyntaxTree.FunctionDefinition	function_definition,
			indent_level = `0`
	)

Definition at line 197 of file MLIRPrinter.py.

References printers.MLIRPrinter.MLIRPrinter.cpp_printer, PipelineBuilder.PipelineBuilder.use_omp, and printers.MLIRPrinter.MLIRPrinter.use_omp.

◆ print_function_type()

printers.MLIRPrinter.MLIRPrinter.print_function_type	(		self,
		SyntaxTree.TFunctionType	type,
			indent_level = `0`
	)

Definition at line 1768 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_if()

printers.MLIRPrinter.MLIRPrinter.print_if	(		self,
		SyntaxTree.If	if_statement,
			indent_level = `0`
	)

Definition at line 289 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_integer()

printers.MLIRPrinter.MLIRPrinter.print_integer	(		self,
		SyntaxTree.Integer	integer,
			indent_level = `0`
	)

Definition at line 257 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_integer_type()

printers.MLIRPrinter.MLIRPrinter.print_integer_type	(		self,
		SyntaxTree.TInteger	type,
			indent_level = `0`
	)

Definition at line 1682 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_long_type()

printers.MLIRPrinter.MLIRPrinter.print_long_type	(		self,
		SyntaxTree.TLong	type,
			indent_level = `0`
	)

Definition at line 1685 of file MLIRPrinter.py.

References printers.CPPPrinter.CPPPrinter.get_indent(), and printers.MLIRPrinter.MLIRPrinter.get_indent().

Here is the call graph for this function:

◆ print_matrix()

printers.MLIRPrinter.MLIRPrinter.print_matrix	(		self,
		SyntaxTree.Matrix	construction,
			indent_level = `0`
	)

Definition at line 1120 of file MLIRPrinter.py.

◆ print_matrix_construction_from_kronecker_product()

printers.MLIRPrinter.MLIRPrinter.print_matrix_construction_from_kronecker_product	(		self,
		SyntaxTree.DataBlockConstructionFromFunction	construction,
			indent_level = `0`
	)