ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

use fast path for sequence generation in old CK (#1993)

Browse Source
This commit is contained in:
Max Podkorytov
2025-03-25 11:28:44 -07:00
committed by GitHub
parent d49abdaa87
commit 1a58522f01
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
include/ck/utility/sequence.hpp
View File

@@ -256,6 +256,18 @@ struct arithmetic_sequence_gen
using type = typename conditional<kHasContent, type0, type1>::type;
};
template <index_t IEnd>
struct arithmetic_sequence_gen<0, IEnd, 1>
{
template <typename T, T... Ints>
struct WrapSequence
{
using type = Sequence<Ints...>;
};
// https://reviews.llvm.org/D13786
using type = typename __make_integer_seq<WrapSequence, index_t, IEnd>::type;
};
// uniform sequence
template <index_t NSize, index_t I>
struct uniform_sequence_gen