/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-7.1.0/include/ck/tensor_operation/gpu/device/impl/split_k_utils.hpp File Reference

• docs-7.1.0
• include
• ck
• tensor_operation
• gpu
• device
• impl

Classes | Namespaces | Functions

split_k_utils.hpp File Reference

#include <numeric>
#include <hip/hip_runtime.h>
#include "ck/utility/env.hpp"
#include "ck/utility/number.hpp"
#include "ck/host_utility/hip_check_error.hpp"
#include "ck/ck.hpp"

Go to the source code of this file.

Classes
struct	ck::tensor_operation::device::DeviceProperties

Namespaces
	ck
	ck::tensor_operation
	ck::tensor_operation::device

Functions
ck::index_t	ck::tensor_operation::device::get_best_occupancy_k_batch_value (int max_occupancy, ck::index_t grid_size)
template<ck::index_t NDimSpatial>
auto	ck::tensor_operation::device::get_bwd_weight_gemm_sizes (const std::array< index_t, NDimSpatial+3 > &a_g_n_k_wos_lengths, const std::array< index_t, NDimSpatial+3 > &e_g_k_c_xs_lengths)
template<ck::index_t MPerBlock, ck::index_t NPerBlock>
ck::index_t	ck::tensor_operation::device::calculate_mn_grid_size (ck::index_t gemmM, ck::index_t gemmN)