/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma.hpp Source File#
warp_gemm_attribute_wmma.hpp
Go to the documentation of this file.
12 // TODO: currently only support 16 bit input, which means only support tr16_b128; will use ADataType
Definition: cluster_descriptor.hpp:13
constexpr bool has_wmma_traits_v
Definition: warp_gemm_attribute_wmma_impl.hpp:133
CK_TILE_HOST bool check_wmma_supported()
Definition: warp_gemm_attribute_wmma.hpp:119
remove_cv_t< std::remove_reference_t< T > > remove_cvref_t
Definition: type_traits.hpp:21
Definition: warp_gemm_attribute_wmma.hpp:16
Definition: warp_gemm_attribute_wmma.hpp:29
Definition: warp_gemm_attribute_wmma.hpp:42
Definition: warp_gemm_attribute_wmma.hpp:55
remove_cvref_t< WarpGemmAttributeWmmaImpl_ > Impl
Definition: warp_gemm_attribute_wmma.hpp:56
CK_TILE_DEVICE void operator()(CVecType &c_vec, const AVecType &a_vec, const BVecType &b_vec, bool_constant< post_nop_ >={}) const
Definition: warp_gemm_attribute_wmma.hpp:84
static constexpr index_t kN
Definition: warp_gemm_attribute_wmma.hpp:67
static constexpr index_t kKPerThread
Definition: warp_gemm_attribute_wmma.hpp:69
typename Impl::ADataType ADataType
Definition: warp_gemm_attribute_wmma.hpp:58
typename Impl::CDataType CDataType
Definition: warp_gemm_attribute_wmma.hpp:60
typename CWarpDstrEncodingTrait< Impl >::type CWarpDstrEncoding
Definition: warp_gemm_attribute_wmma.hpp:80
typename Impl::BVecType BVecType
Definition: warp_gemm_attribute_wmma.hpp:63
static constexpr CK_TILE_HOST_DEVICE auto get_num_of_access()
Definition: warp_gemm_attribute_wmma.hpp:71
typename Impl::BDataType BDataType
Definition: warp_gemm_attribute_wmma.hpp:59
typename Impl::CVecType CVecType
Definition: warp_gemm_attribute_wmma.hpp:64
typename Impl::AVecType AVecType
Definition: warp_gemm_attribute_wmma.hpp:62
static constexpr index_t kK
Definition: warp_gemm_attribute_wmma.hpp:68
static constexpr index_t kM
Definition: warp_gemm_attribute_wmma.hpp:66
CK_TILE_DEVICE CVecType operator()(const AVecType &a_vec, const BVecType &b_vec) const
Definition: warp_gemm_attribute_wmma.hpp:100
typename AWarpDstrEncodingTrait< Impl >::type AWarpDstrEncoding
Definition: warp_gemm_attribute_wmma.hpp:75
typename BWarpDstrEncodingTrait< Impl >::type BWarpDstrEncoding
Definition: warp_gemm_attribute_wmma.hpp:76
Definition: integral_constant.hpp:13
Definition: sequence.hpp:49
Definition: tile_distribution_encoding.hpp:26
Definition: tuple.hpp:192
Definition: arch.hpp:262
Definition: arch.hpp:265