batched_gemm.h File Reference

#include "sdfg/codegen/dispatchers/block_dispatcher.h"
#include "sdfg/data_flow/library_nodes/math/blas/batched_gemm_node.h"

Go to the source code of this file.

Classes
class	sdfg::rocm::blas::BatchedGEMMNodeDispatcher_ROCMBLASWithTransfers
class	sdfg::rocm::blas::BatchedGEMMNodeDispatcher_ROCMBLASWithoutTransfers

Namespaces
namespace	sdfg
namespace	sdfg::rocm
namespace	sdfg::rocm::blas

Functions
void	sdfg::rocm::blas::generate_kernel_batched_gemm (codegen::PrettyPrinter &stream, codegen::LanguageExtension &language_extension, const math::blas::BatchedGEMMNode &node, const std::string &a_name, const std::string &b_name, const std::string &c_name, const std::string &alpha_name, const std::string &beta_name)