tensor_shuffle_utils.hpp File Reference

tensor_shuffle_utils.hpp File Reference#

Composable Kernel: tensor_shuffle_utils.hpp File Reference
tensor_shuffle_utils.hpp File Reference
#include <stdexcept>

Go to the source code of this file.

Namespaces

namespace  ck_tile

Functions

template<typename T>
auto ck_tile::shuffle_aq (const ck_tile::HostTensor< T > *t, int block_aq_k)
template<typename GemmConfig, typename T>
auto ck_tile::shuffle_b (const ck_tile::HostTensor< T > &t)
template<typename GemmConfig, typename T>
auto ck_tile::shuffle_bq_permuteN (const ck_tile::HostTensor< T > &t)
template<typename GemmConfig, typename T>
auto ck_tile::shuffle_b_permuteN (const ck_tile::HostTensor< T > &t)