/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma_impl.hpp Source File#
warp_gemm_attribute_wmma_impl.hpp
Go to the documentation of this file.
Definition: cluster_descriptor.hpp:13
constexpr bool has_wmma_traits_v
Definition: warp_gemm_attribute_wmma_impl.hpp:133
remove_cv_t< std::remove_reference_t< T > > remove_cvref_t
Definition: type_traits.hpp:21
remove_cvref_t< decltype(ck_tile::get_device_arch())> DeviceIp
Definition: warp_gemm_attribute_wmma_impl.hpp:82
Definition: warp_gemm_attribute_wmma_impl.hpp:24
static constexpr index_t kK
Definition: warp_gemm_attribute_wmma_impl.hpp:36
CK_TILE_DEVICE CVecType operator()(const AVecType &a_vec, const BVecType &b_vec) const
Definition: warp_gemm_attribute_wmma_impl.hpp:75
typename Traits::CVecType CVecType
Definition: warp_gemm_attribute_wmma_impl.hpp:31
static constexpr index_t kAMLane
Definition: warp_gemm_attribute_wmma_impl.hpp:42
typename Traits::kCPs2RHssMajor kCPs2RHssMajor
Definition: warp_gemm_attribute_wmma_impl.hpp:58
static constexpr index_t kBNBlock
Definition: warp_gemm_attribute_wmma_impl.hpp:39
typename Traits::BVecType BVecType
Definition: warp_gemm_attribute_wmma_impl.hpp:30
typename Traits::kABYs2RHsMinor kABYs2RHsMinor
Definition: warp_gemm_attribute_wmma_impl.hpp:56
typename Traits::kCYs2RHsMajor kCYs2RHsMajor
Definition: warp_gemm_attribute_wmma_impl.hpp:60
typename Traits::kABPs2RHssMinor kABPs2RHssMinor
Definition: warp_gemm_attribute_wmma_impl.hpp:54
typename Traits::BDataType BDataType
Definition: warp_gemm_attribute_wmma_impl.hpp:26
typename Traits::AVecType AVecType
Definition: warp_gemm_attribute_wmma_impl.hpp:29
typename Traits::kABPs2RHssMajor kABPs2RHssMajor
Definition: warp_gemm_attribute_wmma_impl.hpp:53
typename Traits::kCYs2RHsMinor kCYs2RHsMinor
Definition: warp_gemm_attribute_wmma_impl.hpp:61
static constexpr index_t kAMBlock
Definition: warp_gemm_attribute_wmma_impl.hpp:38
static constexpr index_t kM
Definition: warp_gemm_attribute_wmma_impl.hpp:34
static constexpr index_t kCNLane
Definition: warp_gemm_attribute_wmma_impl.hpp:49
typename Traits::kCPs2RHssMinor kCPs2RHssMinor
Definition: warp_gemm_attribute_wmma_impl.hpp:59
static constexpr index_t kABK0PerLane
Definition: warp_gemm_attribute_wmma_impl.hpp:44
static constexpr index_t kN
Definition: warp_gemm_attribute_wmma_impl.hpp:35
static constexpr index_t kRepeat
Definition: warp_gemm_attribute_wmma_impl.hpp:41
typename Traits::CDataType CDataType
Definition: warp_gemm_attribute_wmma_impl.hpp:27
static constexpr index_t kCM0PerLane
Definition: warp_gemm_attribute_wmma_impl.hpp:50
typename Traits::kABYs2RHsMajor kABYs2RHsMajor
Definition: warp_gemm_attribute_wmma_impl.hpp:55
static constexpr index_t kBNLane
Definition: warp_gemm_attribute_wmma_impl.hpp:43
typename Traits::ADataType ADataType
Definition: warp_gemm_attribute_wmma_impl.hpp:25
static constexpr index_t kABK1PerLane
Definition: warp_gemm_attribute_wmma_impl.hpp:46
static constexpr index_t kCMLane
Definition: warp_gemm_attribute_wmma_impl.hpp:48
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_impl.hpp:65
static constexpr index_t kCM1PerLane
Definition: warp_gemm_attribute_wmma_impl.hpp:51
static constexpr index_t kABKLane
Definition: warp_gemm_attribute_wmma_impl.hpp:45
Definition: warp_gemm_attribute_wmma_impl.hpp:19
Definition: integral_constant.hpp:13
Definition: warp_gemm_attribute_wmma_impl.hpp:112
static constexpr bool value
Definition: warp_gemm_attribute_wmma_impl.hpp:123
static auto test(int) -> decltype(std::declval< typename WmmaTraits< T, AType, BType, CType, warp_m, warp_n, warp_k >::ADataType >(), std::true_type{})
static std::false_type test(...)