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::cuda::blas::BatchedGEMMNodeDispatcher_CUBLASWithTransfers
class	sdfg::cuda::blas::BatchedGEMMNodeDispatcher_CUBLASWithoutTransfers

Namespaces
namespace	sdfg
namespace	sdfg::cuda
namespace	sdfg::cuda::blas

Functions
void	sdfg::cuda::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)