diff --git a/include/ck_tile/core/tensor/slice_tile.hpp b/include/ck_tile/core/tensor/slice_tile.hpp index 7a4ba2eb79..3b696d8cc8 100644 --- a/include/ck_tile/core/tensor/slice_tile.hpp +++ b/include/ck_tile/core/tensor/slice_tile.hpp @@ -76,6 +76,7 @@ set_slice_tile(static_distributed_tensor slice_ends) { using DstDistribution = remove_cvref_t; + using SrcDistribution = remove_cvref_t; constexpr auto sliced_dstr_yidx_ylen = detail::slice_distribution_from_x(DstDistribution{}, slice_begins, slice_ends); @@ -84,9 +85,10 @@ set_slice_tile(static_distributed_tensor(); constexpr auto sliced_y_lengths = sliced_dstr_yidx_ylen.template at<2>(); - static_assert(std::is_same_v, "wrong!"); + static_assert(std::is_same_v, SrcDistribution>, "wrong!"); - dst_tile.SetSlicedThreadData(sliced_y_origins, sliced_y_lengths, src_tile.get_thread_buffer()); + dst_tile.set_y_sliced_thread_data( + sliced_y_origins, sliced_y_lengths, src_tile.get_thread_buffer()); } } // namespace ck_tile