warp_gemm_attribute_wmma_impl.hpp File Reference

• include
• ck_tile
• ops
• gemm
• warp

Classes | Namespaces | Typedefs | Variables

warp_gemm_attribute_wmma_impl.hpp File Reference

#include "ck_tile/core.hpp"
#include "ck_tile/core/config.hpp"

Go to the source code of this file.

Classes
struct	ck_tile::WarpGemmAttributeWmmaImpl< Traits >
struct	ck_tile::has_wmma_traits< Arch, AType, BType, CType, warp_m, warp_n, warp_k >

Namespaces
namespace	ck_tile

Typedefs
using	ck_tile::DeviceIp = remove_cvref_t<decltype(ck_tile::get_device_arch())>
using	ck_tile::WarpGemmAttributeWmmaImpl_f32_16x16x16_f16_f16
using	ck_tile::WarpGemmAttributeWmmaImpl_f32_16x16x16_bf16_bf16
using	ck_tile::WarpGemmAttributeWmmaImpl_i32_16x16x16_i8_i8
using	ck_tile::WarpGemmAttributeWmmaImpl_f32_16x16x16_f8_f8
using	ck_tile::WarpGemmAttributeWmmaImpl_f32_16x16x16_bf8_bf8
using	ck_tile::WarpGemmAttributeWmmaImpl_f32_16x16x16_f8_bf8
using	ck_tile::WarpGemmAttributeWmmaImpl_f32_16x16x16_bf8_f8

Variables
template<typename Arch, typename AType, typename BType, typename CType, index_t warp_m, index_t warp_n, index_t warp_k>
constexpr bool	ck_tile::has_wmma_traits_v