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

adding merge transform

Browse Source 
[ROCm/composable_kernel commit: ca42e9101d]
This commit is contained in:
Chao Liu
2019-09-10 01:53:49 -05:00
parent 399be319a2
commit 17564ecfec
18 changed files with 609 additions and 311 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
composable_kernel/include/utility/tuple.hpp
View File

@@ -3,7 +3,7 @@
#include "integral_constant.hpp"
#include "type.hpp"
#include "Sequence.hpp"
#include "sequence.hpp"
namespace ck {
@@ -114,19 +114,19 @@ __host__ __device__ constexpr auto make_tuple(Xs&&... xs)
namespace detail {
template <typename X, typename F, index_t... Is>
__host__ __device__ constexpr auto transpose_tuple_impl(X& x, F f, Sequence<Is...>)
template <typename F, typename X, index_t... Is>
__host__ __device__ constexpr auto transform_tuple_impl(F f, const X& x, Sequence<Is...>)
{
return make_tuple(f(x.At(Number<Is>{}))...);
}
} // namespace detail
template <typename X, typename F>
__host__ __device__ constexpr auto transpose_tuple(X& x, F f)
template <typename F, typename X>
__host__ __device__ constexpr auto transform_tuple(F f, const X& x)
{
return detail::transpose_tuple_impl(
x, f, typename arithmetic_sequence_gen<0, X::Size(), 1>::type{});
return detail::transform_tuple_impl(
f, x, typename arithmetic_sequence_gen<0, X::Size(), 1>::type{});
}
} // namespace ck