reference_batched_contraction.hpp File Reference

• include
• ck_tile
• host
• reference

Namespaces | Functions

reference_batched_contraction.hpp File Reference

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

Go to the source code of this file.

Namespaces
namespace	ck_tile

Functions
template<typename ADataType, typename BDataType, typename DDataType, typename EDataType, typename AccDataType, typename CDEElementWise>
void	ck_tile::calculate_reference_flat_indexing (const ck_tile::HostTensor< ADataType > &a_full_dims, const ck_tile::HostTensor< BDataType > &b_full_dims, const std::vector< ck_tile::HostTensor< DDataType > > &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)
template<typename ADataType, typename BDataType, typename DDataType, typename EDataType, typename AccDataType, typename CDEElementWise>
void	ck_tile::calculate_reference_multi_dimensional (const HostTensor< ADataType > &a_full_dims, const HostTensor< BDataType > &b_full_dims, const std::vector< HostTensor< DDataType > > &ds_full_dims_host, HostTensor< EDataType > &e_full_dims_host_ref, const std::vector< index_t > &G_dims, const std::vector< index_t > &M_dims, const std::vector< index_t > &N_dims, const std::vector< index_t > &K_dims, const std::vector< index_t > &A_dims, const std::vector< index_t > &B_dims, const std::vector< index_t > &E_dims, const CDEElementWise &cde_elementwise)