#include <sdfg/targets/rocm/blas/dot.h>
|
| | DotNodeDispatcher_ROCMBLASWithTransfers (codegen::LanguageExtension &language_extension, const Function &function, const data_flow::DataFlowGraph &data_flow_graph, const math::blas::DotNode &node) |
| void | dispatch_code (codegen::PrettyPrinter &stream, codegen::PrettyPrinter &globals_stream, codegen::CodeSnippetFactory &library_snippet_factory) override |
| | Generate the operation-specific code.
|
| codegen::InstrumentationInfo | instrumentation_info () const override |
| | Get instrumentation information for this node.
|
| | 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_with_edges (CodegenOutput &out, std::vector< DispatchInput > &inputs, std::vector< DispatchOutput > &outputs) |
◆ DotNodeDispatcher_ROCMBLASWithTransfers()
◆ dispatch_code()
Generate the operation-specific code.
Subclasses override this method to generate the actual library call or inline implementation. This might include:
- BLAS library calls (cblas_dgemm, etc.)
- Standard library calls (memcpy, malloc, etc.)
- Custom inline implementations
- Parameters
-
| stream | Main code stream |
| globals_stream | Global declarations stream |
| library_snippet_factory | Factory for library code snippets |
Reimplemented from sdfg::codegen::LibraryNodeDispatcher.
◆ instrumentation_info()
The documentation for this class was generated from the following file:
- opt/include/sdfg/targets/rocm/blas/dot.h