warp_gemm_attribute_wmma.hpp File Reference

• include
• ck_tile
• ops
• gemm
• warp

Classes | Namespaces | Functions

warp_gemm_attribute_wmma.hpp File Reference

#include "ck_tile/core.hpp"
#include "ck_tile/host/device_prop.hpp"
#include "ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma_impl.hpp"

Go to the source code of this file.

Classes
struct	ck_tile::AWarpDstrEncodingTrait< Impl >
struct	ck_tile::BWarpDstrEncodingTrait< Impl >
struct	ck_tile::CWarpDstrEncodingTrait< Impl >
struct	ck_tile::CTransposedWarpDstrEncodingTrait< Impl >
struct	ck_tile::WarpGemmAttributeWmma< WarpGemmAttributeWmmaImpl_, kTransC >

Namespaces
namespace	ck_tile

Functions
template<typename ADataType, typename BDataType, typename AccDataType, index_t M_Warp_Tile, index_t N_Warp_Tile, index_t K_Warp_Tile>
CK_TILE_HOST bool	ck_tile::check_wmma_supported ()