/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/library/utility/host_gemm.hpp File Reference

• develop
• include
• ck
• library
• utility

Functions

host_gemm.hpp File Reference

#include "host_tensor.hpp"

Go to the source code of this file.

Functions
template<typename AType , typename BType , typename CType , typename AElementwiseOperation , typename BElementwiseOperation , typename CElementwiseOperation >
void	host_gemm_mk_kn_mn (const Tensor< AType > &a_m_k, const Tensor< BType > &b_k_n, Tensor< CType > &c_m_n, const AElementwiseOperation &a_element_op, const BElementwiseOperation &b_element_op, const CElementwiseOperation &c_element_op)

Function Documentation

◆ host_gemm_mk_kn_mn()

template<typename AType , typename BType , typename CType , typename AElementwiseOperation , typename BElementwiseOperation , typename CElementwiseOperation >

void host_gemm_mk_kn_mn	(	const Tensor< AType > &	a_m_k,
		const Tensor< BType > &	b_k_n,
		Tensor< CType > &	c_m_n,
		const AElementwiseOperation &	a_element_op,
		const BElementwiseOperation &	b_element_op,
		const CElementwiseOperation &	c_element_op
	)