/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/block/blockwise_gemm_dlops_v2r2.hpp Source File#
blockwise_gemm_dlops_v2r2.hpp
Go to the documentation of this file.
Definition: ck.hpp:264
__host__ constexpr __device__ auto make_multi_index(Xs &&... xs)
Definition: array_multi_index.hpp:15
__host__ constexpr __device__ auto make_naive_tensor_descriptor_packed(const Tuple< Lengths... > &lengths)
Definition: tensor_descriptor_helper.hpp:101
__host__ constexpr __device__ auto make_merge_transform(const LowLengths &low_lengths)
Definition: multi_index_transform_helper.hpp:55
__host__ constexpr __device__ auto make_single_stage_tensor_adaptor(const Transforms &transforms, LowerDimensionOldTopIdss, UpperDimensionNewTopIdss)
Definition: tensor_adaptor.hpp:429
__host__ constexpr __device__ auto make_pass_through_transform(const LowLength &low_length)
Definition: multi_index_transform_helper.hpp:12
__host__ constexpr __device__ auto make_unmerge_transform(const UpLengths &up_lengths, integral_constant< bool, Use24BitIntegerCalculation >=integral_constant< bool, false >{})
Definition: multi_index_transform_helper.hpp:90
__host__ constexpr __device__ auto transform_tensor_descriptor(const OldTensorDescriptor &old_tensor_desc, const NewTransforms &new_transforms, NewLowerDimensionOldVisibleIdss, NewUpperDimensionNewVisibleIdss)
Definition: tensor_descriptor.hpp:319
__host__ constexpr __device__ auto chain_tensor_adaptors(const TensorAdaptor0 &adaptor0, const TensorAdaptor1 &adaptor1)
Definition: tensor_adaptor.hpp:245
Definition: array.hpp:14
Definition: blockwise_gemm_dlops_v2r2.hpp:47
static constexpr auto b_k_n0_n1_block_desc_
Definition: blockwise_gemm_dlops_v2r2.hpp:143
static constexpr auto I2
Definition: blockwise_gemm_dlops_v2r2.hpp:54
static __device__ CIndex CalculateCM0M1N0N1ThreadOriginOnBlock(index_t thread_id)
Definition: blockwise_gemm_dlops_v2r2.hpp:169
__host__ static constexpr __device__ index_t GetABlockAlignment()
Definition: blockwise_gemm_dlops_v2r2.hpp:192
__host__ static constexpr __device__ auto MakeBKN0N1BlockDescriptor(const BKNBlockDesc &)
Definition: blockwise_gemm_dlops_v2r2.hpp:90
static constexpr auto I0
Definition: blockwise_gemm_dlops_v2r2.hpp:52
static constexpr index_t N101
Definition: blockwise_gemm_dlops_v2r2.hpp:65
MultiIndex< 4 > CIndex
Definition: blockwise_gemm_dlops_v2r2.hpp:50
static constexpr index_t M11
Definition: blockwise_gemm_dlops_v2r2.hpp:67
__host__ static constexpr __device__ auto GetBBlockAlignment()
Definition: blockwise_gemm_dlops_v2r2.hpp:194
static constexpr index_t N11
Definition: blockwise_gemm_dlops_v2r2.hpp:68
static constexpr index_t N
Definition: blockwise_gemm_dlops_v2r2.hpp:59
__host__ static constexpr __device__ auto GetCM0M1N0N1ThreadTensorLengths()
Definition: blockwise_gemm_dlops_v2r2.hpp:137
static constexpr index_t N1
Definition: blockwise_gemm_dlops_v2r2.hpp:71
static constexpr index_t N100
Definition: blockwise_gemm_dlops_v2r2.hpp:62
static constexpr index_t M0
Definition: blockwise_gemm_dlops_v2r2.hpp:73
static constexpr auto I1
Definition: blockwise_gemm_dlops_v2r2.hpp:53
__device__ void Run(const CM0M1N0N1ThreadDesc &, const ABlockBuffer &a_block_buf, const BBlockBuffer &b_block_buf, CThreadBuffer &c_thread_buf) const
Definition: blockwise_gemm_dlops_v2r2.hpp:200
static constexpr index_t K
Definition: blockwise_gemm_dlops_v2r2.hpp:57
static constexpr index_t M100
Definition: blockwise_gemm_dlops_v2r2.hpp:61
static constexpr auto a_k_m0_m1_block_desc_
Definition: blockwise_gemm_dlops_v2r2.hpp:142
static constexpr index_t N0
Definition: blockwise_gemm_dlops_v2r2.hpp:74
__device__ BlockwiseGemmDlops_km_kn_m0m1n0n1_v2r2_pipeline_2x2()
Definition: blockwise_gemm_dlops_v2r2.hpp:146
__host__ static constexpr __device__ auto MakeAKM0M1BlockDescriptor(const AKMBlockDesc &)
Definition: blockwise_gemm_dlops_v2r2.hpp:77
__host__ static constexpr __device__ auto MakeCM0M100M101M11N0N100N101N11ToM0M1N0N1BlockAdaptor()
Definition: blockwise_gemm_dlops_v2r2.hpp:119
static constexpr auto I3
Definition: blockwise_gemm_dlops_v2r2.hpp:55
static constexpr index_t M101
Definition: blockwise_gemm_dlops_v2r2.hpp:64
__host__ static constexpr __device__ auto MakeCM0M100M101M11N0N100N101N11ToMNBlockAdaptor()
Definition: blockwise_gemm_dlops_v2r2.hpp:102
static constexpr index_t M1
Definition: blockwise_gemm_dlops_v2r2.hpp:70
static constexpr index_t M
Definition: blockwise_gemm_dlops_v2r2.hpp:58
Definition: sequence.hpp:43
__device__ void Run(const SrcDesc &, const SrcRefToOriginDisplacement &, const SrcBuffer &src_buf, const DstDesc &, const DstOriginIdx &, DstBuffer &dst_buf) const
Definition: threadwise_tensor_slice_transfer.hpp:1036
Definition: integral_constant.hpp:10
Definition: functional2.hpp:31