Shared storage for the strategy; each composable MMA node has its own dispatcher subclass. More...

Inheritance diagram for sdfg::gpu::GpuMmaNodeDispatcher:

Public Member Functions
	GpuMmaNodeDispatcher (codegen::LanguageExtension &language_extension, const Function &function, const data_flow::DataFlowGraph &data_flow_graph, const data_flow::LibraryNode &node, std::unique_ptr< GpuMmaDispatcherStrategy > strategy)
Public Member Functions inherited from sdfg::codegen::LibraryNodeDispatcher
	LibraryNodeDispatcher (LanguageExtension &language_extension, const Function &function, const data_flow::DataFlowGraph &data_flow_graph, const data_flow::LibraryNode &node)
	Construct a library node dispatcher.
virtual	~LibraryNodeDispatcher ()=default
virtual bool	begin_node (PrettyPrinter &stream)
	Begin code generation for the node.
virtual void	end_node (PrettyPrinter &stream, bool has_declaration)
	End code generation for the node.
virtual void	dispatch (PrettyPrinter &stream, PrettyPrinter &globals_stream, CodeSnippetFactory &library_snippet_factory)
	Dispatch the library node to code.
virtual void	dispatch_code (PrettyPrinter &stream, PrettyPrinter &globals_stream, CodeSnippetFactory &library_snippet_factory)
	Generate the operation-specific code.
virtual void	dispatch_code_with_edges (CodegenOutput &out, std::vector< DispatchInput > &inputs, std::vector< DispatchOutput > &outputs)
virtual InstrumentationInfo	instrumentation_info () const
	Get instrumentation information for this node.

Protected Attributes
std::unique_ptr< GpuMmaDispatcherStrategy >	strategy_
Protected Attributes inherited from sdfg::codegen::LibraryNodeDispatcher
LanguageExtension &	language_extension_
	Language extension for code generation.
const Function &	function_
	Function context.
const data_flow::DataFlowGraph &	data_flow_graph_
	Containing dataflow graph.
const data_flow::LibraryNode &	node_
	Library node being dispatched.

Additional Inherited Members
Protected Member Functions inherited from sdfg::codegen::LibraryNodeDispatcher
void	require_locally_modifiable_var (CodegenOutput &out, DispatchInput &input) const
	Ensure that [input] points to a local copy that can be changed without modifying the actual source of the input Should only be needed to ensure that legacy [dispatch_code] code can act as before, where it always was a copy.
void	pre_allocate_output (CodegenOutput &out, DispatchOutput &output, const std::string &var_name) const
	Declare a variable of the type of the output and with the name "var_name".
void	register_output (DispatchOutput &output, const std::string &result_identifier) const
	The dispatcher as created an output that is identified by [result_identifier].
void	copy_output (CodegenOutput &out, const DispatchOutput &output, const data_flow::Memlet &oedge) const
	This is a temporary & hopefully short-lived workaround.

Detailed Description

Shared storage for the strategy; each composable MMA node has its own dispatcher subclass.

Constructor & Destructor Documentation

sdfg::gpu::GpuMmaNodeDispatcher::GpuMmaNodeDispatcher	(	codegen::LanguageExtension &	language_extension,
		const Function &	function,
		const data_flow::DataFlowGraph &	data_flow_graph,
		const data_flow::LibraryNode &	node,
		std::unique_ptr< GpuMmaDispatcherStrategy >	strategy )

inline

std::unique_ptr<GpuMmaDispatcherStrategy> sdfg::gpu::GpuMmaNodeDispatcher::strategy_

protected

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