/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck_tile/ops/gemm/warp/warp_gemm_impl.hpp Source File#
warp_gemm_impl.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:20
typename impl::ext_vector< T, N >::type ext_vector_t
Definition: vector_type.hpp:54
constexpr CK_TILE_HOST_DEVICE auto make_static_tile_distribution(StaticTileDistributionEncoding_)
Definition: tile_distribution.hpp:498
Definition: warp_gemm_impl.hpp:11
CK_TILE_DEVICE void operator()(CTensor &c, const ATensor &a, const BTensor &b, number< i_subk >, bool_constant< post_nop_ >={}) const
Definition: warp_gemm_impl.hpp:72
static constexpr CK_TILE_HOST_DEVICE auto get_num_of_access()
Definition: warp_gemm_impl.hpp:39
typename WarpGemmAttribute::CWarpDstrEncoding CWarpDstrEncoding
Definition: warp_gemm_impl.hpp:29
CK_TILE_DEVICE auto operator()(const ATensor &a, const BTensor &b) const
Definition: warp_gemm_impl.hpp:95
typename WarpGemmAttribute::BWarpDstrEncoding BWarpDstrEncoding
Definition: warp_gemm_impl.hpp:28
remove_cvref_t< decltype(make_static_tile_distribution(BWarpDstrEncoding{}))> BWarpDstr
Definition: warp_gemm_impl.hpp:32
static constexpr index_t kKPerThread
The number of elements in K dimension processed by single thread in wavefront.
Definition: warp_gemm_impl.hpp:21
typename WarpGemmAttribute::AWarpDstrEncoding AWarpDstrEncoding
Definition: warp_gemm_impl.hpp:27
CK_TILE_DEVICE void operator()(CTensor &c, const ATensor &a, const BTensor &b, bool_constant< post_nop_ >={}) const
Definition: warp_gemm_impl.hpp:46
typename WarpGemmAttribute::CDataType CDataType
Definition: warp_gemm_impl.hpp:25
typename WarpGemmAttribute::ADataType ADataType
Definition: warp_gemm_impl.hpp:23
remove_cvref_t< decltype(make_static_tile_distribution(CWarpDstrEncoding{}))> CWarpDstr
Definition: warp_gemm_impl.hpp:33
remove_cvref_t< WarpGemmAttribute_ > WarpGemmAttribute
Definition: warp_gemm_impl.hpp:12
remove_cvref_t< decltype(make_static_tile_distribution(AWarpDstrEncoding{}))> AWarpDstr
Definition: warp_gemm_impl.hpp:31
static_distributed_tensor< CDataType, CWarpDstr > CWarpTensor
Definition: warp_gemm_impl.hpp:37
typename WarpGemmAttribute::BDataType BDataType
Definition: warp_gemm_impl.hpp:24
Definition: integral_constant.hpp:13
Definition: static_distributed_tensor.hpp:21