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

added ReorderGiveOld2New() in Sequence and ConstantTensorDescriptor

Browse Source
This commit is contained in:
Chao Liu
2019-08-06 18:48:05 -05:00
parent fdcfae3a62
commit 0271338ed4
3 changed files with 45 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
composable_kernel/include/tensor_operation/threadwise_generic_tensor_slice_copy.hpp
View File

@@ -74,8 +74,8 @@ __device__ void threadwise_generic_tensor_slice_copy_v1(
constexpr auto data_multi_id_in_access_order =
access_multi_id.Modify(Number<nDim - 1>{}, Number<itmp>{});
constexpr auto data_multi_id = reorder_array_given_old2new(
sequence2array(data_multi_id_in_access_order), DimAccessOrder{});
constexpr auto data_multi_id =
data_multi_id_in_access_order.ReorderGivenOld2New(DimAccessOrder{});
const index_t src_index =
SrcDesc::GetOffsetFromMultiIndex(src_multi_id_begin + data_multi_id);