docc 0.8.3 (0d44999b)
C++ API Reference
Loading...
Searching...
No Matches
sdfg::cuda::blas::DotNodeDispatcher_CUBLASWithoutTransfers Class Reference

#include <sdfg/targets/cuda/blas/dot.h>

Inheritance diagram for sdfg::cuda::blas::DotNodeDispatcher_CUBLASWithoutTransfers:
[legend]

Public Member Functions

 DotNodeDispatcher_CUBLASWithoutTransfers (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.
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_with_edges (CodegenOutput &out, std::vector< DispatchInput > &inputs, std::vector< DispatchOutput > &outputs)

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.
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.

Constructor & Destructor Documentation

◆ DotNodeDispatcher_CUBLASWithoutTransfers()

sdfg::cuda::blas::DotNodeDispatcher_CUBLASWithoutTransfers::DotNodeDispatcher_CUBLASWithoutTransfers ( codegen::LanguageExtension & language_extension,
const Function & function,
const data_flow::DataFlowGraph & data_flow_graph,
const math::blas::DotNode & node )

Member Function Documentation

◆ dispatch_code()

void sdfg::cuda::blas::DotNodeDispatcher_CUBLASWithoutTransfers::dispatch_code ( codegen::PrettyPrinter & stream,
codegen::PrettyPrinter & globals_stream,
codegen::CodeSnippetFactory & library_snippet_factory )
overridevirtual

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
streamMain code stream
globals_streamGlobal declarations stream
library_snippet_factoryFactory for library code snippets

Reimplemented from sdfg::codegen::LibraryNodeDispatcher.

◆ instrumentation_info()

codegen::InstrumentationInfo sdfg::cuda::blas::DotNodeDispatcher_CUBLASWithoutTransfers::instrumentation_info ( ) const
overridevirtual

Get instrumentation information for this node.

Returns
Instrumentation information

Reimplemented from sdfg::codegen::LibraryNodeDispatcher.


The documentation for this class was generated from the following file:
  • opt/include/sdfg/targets/cuda/blas/dot.h