/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_mfma.hpp Source File#
warp_gemm_attribute_mfma.hpp
Go to the documentation of this file.
Definition: cluster_descriptor.hpp:13
remove_cv_t< std::remove_reference_t< T > > remove_cvref_t
Definition: type_traits.hpp:21
typename impl::ext_vector< T, N >::type ext_vector_t
Definition: vector_type.hpp:83
Definition: warp_gemm_attribute_mfma.hpp:23
static constexpr auto get_warp_dstr_encoding()
Definition: warp_gemm_attribute_mfma.hpp:48
static constexpr index_t kK
Definition: warp_gemm_attribute_mfma.hpp:38
typename Impl::BDataType BDataType
Definition: warp_gemm_attribute_mfma.hpp:29
static constexpr index_t kCMLane
Definition: warp_gemm_attribute_mfma.hpp:40
typename Impl::AVecType AVecType
Definition: warp_gemm_attribute_mfma.hpp:32
typename Impl::CVecType CVecType
Definition: warp_gemm_attribute_mfma.hpp:34
decltype(get_warp_dstr_encoding< Impl::kAMLane >()) AWarpDstrEncoding
Definition: warp_gemm_attribute_mfma.hpp:74
static constexpr auto AttrNumAccess
Definition: warp_gemm_attribute_mfma.hpp:25
typename Impl::CDataType CDataType
Definition: warp_gemm_attribute_mfma.hpp:30
CK_TILE_DEVICE CVecType operator()(const AVecType &a_vec, const BVecType &b_vec) const
Definition: warp_gemm_attribute_mfma.hpp:97
static constexpr CK_TILE_HOST_DEVICE auto get_num_of_access()
Definition: warp_gemm_attribute_mfma.hpp:42
static constexpr auto AttrNumAccessV
Definition: warp_gemm_attribute_mfma.hpp:26
static constexpr index_t kM
Definition: warp_gemm_attribute_mfma.hpp:36
remove_cvref_t< WarpGemmAttributeMfmaImpl_ > Impl
Definition: warp_gemm_attribute_mfma.hpp:24
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_mfma.hpp:88
static constexpr index_t kKPerThread
Definition: warp_gemm_attribute_mfma.hpp:39
static constexpr index_t kN
Definition: warp_gemm_attribute_mfma.hpp:37
typename Impl::ADataType ADataType
Definition: warp_gemm_attribute_mfma.hpp:28
decltype(get_warp_dstr_encoding< Impl::kBNLane >()) BWarpDstrEncoding
Definition: warp_gemm_attribute_mfma.hpp:75
typename Impl::BVecType BVecType
Definition: warp_gemm_attribute_mfma.hpp:33
Definition: warp_gemm_attribute_mfma.hpp:845
remove_cvref_t< WarpGemmAttributeMfmaImpl_ > Impl
Definition: warp_gemm_attribute_mfma.hpp:846
CK_TILE_DEVICE void operator()(CVecType &c_vec, const AVecType &a_vec, const BVecType &b_vec, number< iKIter >, bool_constant< post_nop_ >={}) const
Definition: warp_gemm_attribute_mfma.hpp:919
static constexpr index_t kN
Definition: warp_gemm_attribute_mfma.hpp:859
static constexpr index_t kM
Definition: warp_gemm_attribute_mfma.hpp:858
static constexpr index_t kK
Definition: warp_gemm_attribute_mfma.hpp:860
CK_TILE_DEVICE CVecType operator()(const AVecType &a_vec, const BVecType &b_vec) const
Definition: warp_gemm_attribute_mfma.hpp:941
static constexpr index_t kKPerThread
Definition: warp_gemm_attribute_mfma.hpp:861
ext_vector_t< BDataType, vector_traits< typename Impl::BVecType >::vector_size *kKIter > BVecType
Definition: warp_gemm_attribute_mfma.hpp:855
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_mfma.hpp:900
typename Impl::BDataType BDataType
Definition: warp_gemm_attribute_mfma.hpp:849
typename Impl::CVecType CVecType
Definition: warp_gemm_attribute_mfma.hpp:856
static constexpr index_t SFactor
Definition: warp_gemm_attribute_mfma.hpp:862
static constexpr CK_TILE_HOST_DEVICE auto get_num_of_access()
Definition: warp_gemm_attribute_mfma.hpp:864
typename Impl::ADataType ADataType
Definition: warp_gemm_attribute_mfma.hpp:848
typename Impl::CDataType CDataType
Definition: warp_gemm_attribute_mfma.hpp:850
ext_vector_t< ADataType, vector_traits< typename Impl::AVecType >::vector_size *kKIter > AVecType
Definition: warp_gemm_attribute_mfma.hpp:853
Definition: warp_gemm_attribute_mfma.hpp:701
typename Impl::BDataType ADataType
Definition: warp_gemm_attribute_mfma.hpp:705
typename Impl::ADataType BDataType
Definition: warp_gemm_attribute_mfma.hpp:706
static constexpr index_t kKPerThread
Definition: warp_gemm_attribute_mfma.hpp:718
ext_vector_t< ADataType, vector_traits< typename Impl::AVecType >::vector_size *kKIter > AVecType
Definition: warp_gemm_attribute_mfma.hpp:710
static constexpr index_t SFactor
Definition: warp_gemm_attribute_mfma.hpp:719
static constexpr index_t kK
Definition: warp_gemm_attribute_mfma.hpp:717
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_mfma.hpp:779
typename Impl::CVecType CVecType
Definition: warp_gemm_attribute_mfma.hpp:713
remove_cvref_t< WarpGemmAttributeMfmaImpl_ > Impl
Definition: warp_gemm_attribute_mfma.hpp:702
typename Impl::CDataType CDataType
Definition: warp_gemm_attribute_mfma.hpp:707
static constexpr index_t kN
Definition: warp_gemm_attribute_mfma.hpp:716
CK_TILE_DEVICE CVecType operator()(const AVecType &a_vec, const BVecType &b_vec) const
Definition: warp_gemm_attribute_mfma.hpp:820
CK_TILE_DEVICE void operator()(CVecType &c_vec, const AVecType &a_vec, const BVecType &b_vec, number< iKIter >, bool_constant< post_nop_ >={}) const
Definition: warp_gemm_attribute_mfma.hpp:798
ext_vector_t< BDataType, vector_traits< typename Impl::BVecType >::vector_size *kKIter > BVecType
Definition: warp_gemm_attribute_mfma.hpp:712
static constexpr index_t kM
Definition: warp_gemm_attribute_mfma.hpp:715
static constexpr CK_TILE_HOST_DEVICE auto get_num_of_access()
Definition: warp_gemm_attribute_mfma.hpp:721
Definition: warp_gemm_attribute_mfma.hpp:551
typename Impl::CVecType CVecType
Definition: warp_gemm_attribute_mfma.hpp:564
static constexpr CK_TILE_DEVICE auto get_cwarp_dstr_encoding()
Definition: warp_gemm_attribute_mfma.hpp:588
ext_vector_t< BDataType, vector_traits< typename Impl::BVecType >::vector_size *kKIter > BVecType
Definition: warp_gemm_attribute_mfma.hpp:563
static constexpr index_t kKPerThread
Definition: warp_gemm_attribute_mfma.hpp:569
typename Impl::CDataType CDataType
Definition: warp_gemm_attribute_mfma.hpp:558
static constexpr CK_TILE_HOST_DEVICE auto get_num_of_access()
Definition: warp_gemm_attribute_mfma.hpp:571
typename Impl::BDataType ADataType
Definition: warp_gemm_attribute_mfma.hpp:556
static constexpr index_t kM
Definition: warp_gemm_attribute_mfma.hpp:566
static constexpr CK_TILE_DEVICE auto get_awarp_dstr_encoding()
Definition: warp_gemm_attribute_mfma.hpp:576
static constexpr index_t kK
Definition: warp_gemm_attribute_mfma.hpp:568
CK_TILE_DEVICE void operator()(CVecType &c_vec, const AVecType &a_vec, const BVecType &b_vec, number< iKIter >, bool_constant< post_nop_ >={}) const
Definition: warp_gemm_attribute_mfma.hpp:654
static constexpr index_t kN
Definition: warp_gemm_attribute_mfma.hpp:567
decltype(get_awarp_dstr_encoding()) AWarpDstrEncoding
Definition: warp_gemm_attribute_mfma.hpp:626
CK_TILE_DEVICE CVecType operator()(const AVecType &a_vec, const BVecType &b_vec) const
Definition: warp_gemm_attribute_mfma.hpp:676
decltype(get_bwarp_dstr_encoding()) BWarpDstrEncoding
Definition: warp_gemm_attribute_mfma.hpp:628
static constexpr CK_TILE_DEVICE auto get_bwarp_dstr_encoding()
Definition: warp_gemm_attribute_mfma.hpp:582
decltype(get_cwarp_dstr_encoding()) CWarpDstrEncoding
Definition: warp_gemm_attribute_mfma.hpp:630
remove_cvref_t< WarpGemmAttributeMfmaImpl_ > Impl
Definition: warp_gemm_attribute_mfma.hpp:552
static constexpr auto AttrNumAccess
Definition: warp_gemm_attribute_mfma.hpp:553
ext_vector_t< ADataType, vector_traits< typename Impl::AVecType >::vector_size *kKIter > AVecType
Definition: warp_gemm_attribute_mfma.hpp:561
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_mfma.hpp:634
typename Impl::ADataType BDataType
Definition: warp_gemm_attribute_mfma.hpp:557
Definition: warp_gemm_attribute_mfma.hpp:107
static constexpr auto AttrNumAccess
Definition: warp_gemm_attribute_mfma.hpp:111
decltype(get_bwarp_dstr_encoding()) BWarpDstrEncoding
Definition: warp_gemm_attribute_mfma.hpp:297
static constexpr CK_TILE_DEVICE auto get_cwarp_dstr_encoding()
Definition: warp_gemm_attribute_mfma.hpp:257
static constexpr CK_TILE_DEVICE auto get_bwarp_dstr_encoding()
Definition: warp_gemm_attribute_mfma.hpp:195
typename Impl::CVecType CVecType
Definition: warp_gemm_attribute_mfma.hpp:122
typename Impl::BDataType BDataType
Definition: warp_gemm_attribute_mfma.hpp:115
ext_vector_t< ADataType, vector_traits< typename Impl::AVecType >::vector_size *kKIter > AVecType
Definition: warp_gemm_attribute_mfma.hpp:119
CK_TILE_DEVICE void operator()(CVecType &c_vec, const AVecType &a_vec, const BVecType &b_vec, number< iKIter >, bool_constant< post_nop_ >={}) const
Definition: warp_gemm_attribute_mfma.hpp:322
decltype(get_cwarp_dstr_encoding()) CWarpDstrEncoding
Definition: warp_gemm_attribute_mfma.hpp:299
ext_vector_t< BDataType, vector_traits< typename Impl::BVecType >::vector_size *kKIter > BVecType
Definition: warp_gemm_attribute_mfma.hpp:121
CK_TILE_DEVICE CVecType operator()(const AVecType &a_vec, const BVecType &b_vec) const
Definition: warp_gemm_attribute_mfma.hpp:344
static constexpr CK_TILE_HOST_DEVICE auto get_num_of_access()
Definition: warp_gemm_attribute_mfma.hpp:129
static constexpr CK_TILE_DEVICE auto get_awarp_dstr_encoding()
Definition: warp_gemm_attribute_mfma.hpp:134
typename Impl::CDataType CDataType
Definition: warp_gemm_attribute_mfma.hpp:116
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_mfma.hpp:303
typename Impl::ADataType ADataType
Definition: warp_gemm_attribute_mfma.hpp:114
static constexpr index_t kM
Definition: warp_gemm_attribute_mfma.hpp:124
static constexpr index_t kK
Definition: warp_gemm_attribute_mfma.hpp:126
static constexpr auto AttrNumAccessV
Definition: warp_gemm_attribute_mfma.hpp:112
decltype(get_awarp_dstr_encoding()) AWarpDstrEncoding
Definition: warp_gemm_attribute_mfma.hpp:295
static constexpr index_t kN
Definition: warp_gemm_attribute_mfma.hpp:125
remove_cvref_t< WarpGemmAttributeMfmaImpl_ > Impl
Definition: warp_gemm_attribute_mfma.hpp:110
static constexpr index_t kKPerThread
Definition: warp_gemm_attribute_mfma.hpp:127
Definition: warp_gemm_attribute_mfma.hpp:455
static constexpr index_t kN
Definition: warp_gemm_attribute_mfma.hpp:467
static constexpr index_t SFactor
Definition: warp_gemm_attribute_mfma.hpp:470
typename Impl::BVecType AVecType
Definition: warp_gemm_attribute_mfma.hpp:462
typename Impl::AVecType BVecType
Definition: warp_gemm_attribute_mfma.hpp:463
typename Impl::ADataType BDataType
Definition: warp_gemm_attribute_mfma.hpp:459
CK_TILE_DEVICE CVecType operator()(const AVecType &a_vec, const BVecType &b_vec) const
Definition: warp_gemm_attribute_mfma.hpp:540
remove_cvref_t< WarpGemmAttributeMfmaImpl_ > Impl
Definition: warp_gemm_attribute_mfma.hpp:456
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_mfma.hpp:530
static constexpr index_t kKPerThread
Definition: warp_gemm_attribute_mfma.hpp:469
typename Impl::CDataType CDataType
Definition: warp_gemm_attribute_mfma.hpp:460
static constexpr index_t kM
Definition: warp_gemm_attribute_mfma.hpp:466
typename Impl::BDataType ADataType
Definition: warp_gemm_attribute_mfma.hpp:458
static constexpr CK_TILE_HOST_DEVICE auto get_num_of_access()
Definition: warp_gemm_attribute_mfma.hpp:472
static constexpr index_t kK
Definition: warp_gemm_attribute_mfma.hpp:468
typename Impl::CVecType CVecType
Definition: warp_gemm_attribute_mfma.hpp:464
Definition: warp_gemm_attribute_mfma.hpp:371
CK_TILE_DEVICE CVecType operator()(const AVecType &a_vec, const BVecType &b_vec) const
Definition: warp_gemm_attribute_mfma.hpp:446
static constexpr index_t kCMLane
Definition: warp_gemm_attribute_mfma.hpp:388
typename Impl::BDataType ADataType
Definition: warp_gemm_attribute_mfma.hpp:376
typename Impl::AVecType BVecType
Definition: warp_gemm_attribute_mfma.hpp:381
static constexpr index_t kKPerThread
Definition: warp_gemm_attribute_mfma.hpp:387
static constexpr CK_TILE_HOST_DEVICE auto get_num_of_access()
Definition: warp_gemm_attribute_mfma.hpp:390
decltype(get_warp_dstr_encoding< Impl::kAMLane >()) BWarpDstrEncoding
Definition: warp_gemm_attribute_mfma.hpp:423
typename Impl::BVecType AVecType
Definition: warp_gemm_attribute_mfma.hpp:380
static constexpr index_t kK
Definition: warp_gemm_attribute_mfma.hpp:386
typename Impl::ADataType BDataType
Definition: warp_gemm_attribute_mfma.hpp:377
static constexpr auto AttrNumAccessV
Definition: warp_gemm_attribute_mfma.hpp:374
static constexpr auto get_warp_dstr_encoding()
Definition: warp_gemm_attribute_mfma.hpp:396
typename Impl::CDataType CDataType
Definition: warp_gemm_attribute_mfma.hpp:378
static constexpr auto AttrNumAccess
Definition: warp_gemm_attribute_mfma.hpp:373
typename Impl::CVecType CVecType
Definition: warp_gemm_attribute_mfma.hpp:382
static constexpr index_t kN
Definition: warp_gemm_attribute_mfma.hpp:385
remove_cvref_t< WarpGemmAttributeMfmaImpl_ > Impl
Definition: warp_gemm_attribute_mfma.hpp:372
decltype(get_warp_dstr_encoding< Impl::kBNLane >()) AWarpDstrEncoding
Definition: warp_gemm_attribute_mfma.hpp:422
static constexpr index_t kM
Definition: warp_gemm_attribute_mfma.hpp:384
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_mfma.hpp:436
Definition: integral_constant.hpp:13
Definition: sequence.hpp:49
Definition: functional.hpp:43
Definition: debug.hpp:67
Definition: tile_distribution_encoding.hpp:26
Definition: tuple.hpp:192