ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 03:37:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small concepts clean-up.

Browse Source
This commit is contained in:
Ville Pietilä
2026-01-14 06:22:46 -05:00
parent 07608d1a86
commit 3f7b250d33
1 changed files with 3 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
experimental/builder/include/ck_tile/builder/conv_algorithm_concepts.hpp
View File

@@ -38,23 +38,15 @@ concept WarpGemmDescriptor = requires(T t) {
{ t.gemm_n_iters_per_wave } -> SizeType;
};
// Concept for parameter that describe block GEMM problem.
// Concept for parameters that describe the GEMM pipeline.
template <typename T>
concept GemmPipelineDescriptor = requires(T t) {
{ t.num_conv_groups_to_merge } -> SizeType;
{ t.num_gemm_k_prefetch_stages } -> SizeType;
{ t.pipeline_version } -> std::convertible_to<PipelineVersion>;
{ t.scheduler } -> std::convertible_to<PipelineScheduler>;
};
// Concept for parameters that describe a gridwise WMMA GEMM problem.
template <typename T>
concept GridwiseWmmaGemmDescriptor = requires(T t) {
{ t.k1 } -> SizeType;
{ t.m_per_wmma } -> SizeType;
{ t.n_per_wmma } -> SizeType;
{ t.m_wmma_per_wave } -> SizeType;
{ t.n_wmma_per_wave } -> SizeType;
};
// Concept for vectorized data transfer for convolution input tensors.
template <typename T>
concept InputTileThreadClusterDescriptor3D = requires(T t) {