sdfg::cuda::blas Namespace Reference

Classes
class	BatchedGEMMNodeDispatcher_CUBLASWithTransfers
class	BatchedGEMMNodeDispatcher_CUBLASWithoutTransfers
class	DotNodeDispatcher_CUBLASWithTransfers
class	DotNodeDispatcher_CUBLASWithoutTransfers
class	GEMMNodeDispatcher_CUBLASWithTransfers
class	GEMMNodeDispatcher_CUBLASWithoutTransfers

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	cublas_error_checking (codegen::PrettyPrinter &stream, const codegen::LanguageExtension &language_extension, const std::string &status_variable)

Function Documentation

create_blas_handle()

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

cublas_error_checking()

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

destroy_blas_handle()

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

generate_kernel_batched_gemm()

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 )

generate_kernel_gemm()

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

setup_blas_handle()

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