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

[CK TILE] only use trivial_array in sequence.hpp

Browse Source
This commit is contained in:
Cong Ma
2026-01-27 13:03:48 -05:00
parent 82ef9432cc
commit 0acf5e439f
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
include/ck_tile/core/container/sequence.hpp
View File

@@ -1143,13 +1143,16 @@ struct sorted_sequence_histogram<h_idx, sequence<x>, sequence<r, rs...>>
};
} // namespace detail
template <typename, index_t>
struct array; // declare for later use (array->seq utility)
// SeqSortedSamples: <0, 2, 3, 5, 7>, SeqRange: <0, 3, 6, 9> -> SeqHistogram : <2, 2, 1>
template <typename SeqSortedSamples, index_t r, index_t... rs>
CK_TILE_HOST_DEVICE constexpr auto histogram_sorted_sequence(SeqSortedSamples, sequence<r, rs...>)
{
constexpr auto bins = sizeof...(rs); // or categories
constexpr auto histogram = [&]() {
trivial_array<index_t, bins> h{0}; // make sure this can clear all element to zero
array<index_t, bins> h{0}; // make sure this can clear all element to zero
detail::sorted_sequence_histogram<0, SeqSortedSamples, sequence<rs...>>{}(h);
return h;
}();
@@ -1162,7 +1165,7 @@ CK_TILE_HOST_DEVICE constexpr auto generate_array(F&& f, number<N>)
{
using T = remove_cvref_t<decltype(f(number<0>{}))>;
return unpack([&f](auto&&... is) { return trivial_array<T, N>{f(is)...}; },
return unpack([&f](auto&&... is) { return array<T, N>{f(is)...}; },
typename arithmetic_sequence_gen<0, N, 1>::type{});
}

2
include/ck_tile/core/container/trivial_array.hpp
View File

@@ -16,6 +16,6 @@ struct trivial_array
constexpr const T& operator[](index_t i) const { return data[i]; }
constexpr T& operator[](index_t i) { return data[i]; }
constexpr index_t size() const { return N; }
constexpr static index_t size() { return N; }
};
} // namespace ck_tile