sdfg::rocm::blas Namespace Reference

Classes
class	BatchedGEMMNodeDispatcher_ROCMBLASWithTransfers
class	BatchedGEMMNodeDispatcher_ROCMBLASWithoutTransfers
class	DotNodeDispatcher_ROCMBLASWithTransfers
class	DotNodeDispatcher_ROCMBLASWithoutTransfers
class	GEMMNodeDispatcher_ROCMBLASWithTransfers
class	GEMMNodeDispatcher_ROCMBLASWithoutTransfers

Functions
void	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)
void	generate_kernel_gemm (codegen::PrettyPrinter &stream, codegen::LanguageExtension &language_extension, const math::blas::GEMMNode &gemm_node)
void	create_blas_handle (codegen::PrettyPrinter &stream, const codegen::LanguageExtension &language_extension)
void	destroy_blas_handle (codegen::PrettyPrinter &stream, const codegen::LanguageExtension &language_extension)
void	setup_blas_handle (codegen::CodeSnippetFactory &factory, const codegen::LanguageExtension &language_extension)
void	rocmblas_error_checking (codegen::PrettyPrinter &stream, const codegen::LanguageExtension &language_extension, const std::string &status_variable)

Function Documentation

create_blas_handle()

void sdfg::rocm::blas::create_blas_handle	(	codegen::PrettyPrinter &	stream,
		const codegen::LanguageExtension &	language_extension )

destroy_blas_handle()

void sdfg::rocm::blas::destroy_blas_handle	(	codegen::PrettyPrinter &	stream,
		const codegen::LanguageExtension &	language_extension )

generate_kernel_batched_gemm()

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 )

generate_kernel_gemm()

void sdfg::rocm::blas::generate_kernel_gemm	(	codegen::PrettyPrinter &	stream,
		codegen::LanguageExtension &	language_extension,
		const math::blas::GEMMNode &	gemm_node )

rocmblas_error_checking()

void sdfg::rocm::blas::rocmblas_error_checking	(	codegen::PrettyPrinter &	stream,
		const codegen::LanguageExtension &	language_extension,
		const std::string &	status_variable )

setup_blas_handle()

void sdfg::rocm::blas::setup_blas_handle	(	codegen::CodeSnippetFactory &	factory,
		const codegen::LanguageExtension &	language_extension )