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

bug fix: BlockwiseGenericTensorSliceCopy_v2::MoveDstSlicingWindow

Browse Source
This commit is contained in:
Chao Liu
2019-08-15 15:12:13 -05:00
parent 86ceded98b
commit 08bf57b01c
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -417,6 +417,8 @@ struct BlockwiseGenericTensorSliceCopy_v1
}
};
// Slice a (normal or merged) tensor, and copy it into another (normal or merged) tensor
// memory layout (ordering of dimensions) can be different between src and dst.
template <index_t BlockSize,
class SrcDesc,
class DstDesc,
@@ -510,7 +512,7 @@ struct BlockwiseGenericTensorSliceCopy_v2
MoveDstSlicingWindow(T step_sizes,
integral_constant<bool, PositiveDirection> positive_direction)
{
mThreadwiseLoad.MoveDstSlicingWindow(step_sizes, positive_direction);
mThreadwiseStore.MoveDstSlicingWindow(step_sizes, positive_direction);
}
private: