/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/host/reference/reference_batched_contraction.hpp File Reference

• develop
• include
• ck_tile
• host
• reference

Classes | Namespaces | Functions

reference_batched_contraction.hpp File Reference

#include <cstdlib>
#include <thread>
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"

Go to the source code of this file.

Classes
struct	ck_tile::ApplyCDEElementWise< EDataType, AccDataType, CDEElementWise >
struct	ck_tile::ExtractDValues< DDataType, NumDTensor, std::index_sequence< Is... > >

Namespaces
	ck_tile

Functions
template<typename ADataType , typename BDataType , typename DDataType , typename EDataType , typename AccDataType , typename CDEElementWise , ck_tile::index_t NumDTensor>
void	ck_tile::compute_reference_batched_contraction (const ck_tile::HostTensor< ADataType > &a_full_dims, const ck_tile::HostTensor< BDataType > &b_full_dims, const std::array< ck_tile::HostTensor< DDataType >, NumDTensor > &ds_full_dims_host, ck_tile::HostTensor< EDataType > &e_full_dims_host_ref, ck_tile::index_t G_total, ck_tile::index_t M_total, ck_tile::index_t N_total, ck_tile::index_t K_total, const CDEElementWise &cde_elementwise, const std::vector< ck_tile::index_t > &G_dims, const std::vector< ck_tile::index_t > &M_dims, const std::vector< ck_tile::index_t > &N_dims, const std::vector< ck_tile::index_t > &K_dims)