Commit 9a54fbd8 authored by Jing Zhang's avatar Jing Zhang
Browse files

clean code

parent be11c5dd
......@@ -141,14 +141,12 @@ struct BlockwiseGenericTensorSliceCopy_v5
private:
using ThreadBufferDesc = decltype(make_native_tensor_descriptor_packed(ThreadSliceLengths{}));
using ThreadBufferType = decltype(GetRegBuffer<float, GetThreadBufferSize()>());
using ThreadwiseCopy = ThreadwiseGenericTensorSliceCopy_v5<BlockSrcDesc,
BlockDstDesc,
ThreadSliceLengths,
SrcDimAccessOrder,
DstDimAccessOrder,
ThreadBufferType,
SrcVectoReadDim,
DstVectorWriteDim,
SrcDataPerRead,
......
......@@ -19,7 +19,6 @@ template <typename SrcDesc,
typename SliceLengths,
typename SrcDimAccessOrder,
typename DstDimAccessOrder,
typename BufferVectorType,
index_t SrcVectorReadDim,
index_t DstVectorWriteDim,
index_t SrcDataPerRead,
......@@ -35,6 +34,10 @@ struct ThreadwiseGenericTensorSliceCopy_v5
static constexpr index_t ThreadBufferSize = ThreadBufferDesc::GetElementSpace();
using ThreadBufferType = decltype(GetRegBuffer<float, ThreadBufferSize>());
ThreadBufferType thread_buff;
static constexpr index_t nDim = SliceLengths::Size();
using Index = MultiIndex<nDim>;
......@@ -253,8 +256,6 @@ struct ThreadwiseGenericTensorSliceCopy_v5
.Else([&](auto) { mDstSliceOrigin -= step_sizes; });
}
BufferVectorType thread_buff;
private:
SrcCoord mSrcSliceOrigin;
DstCoord mDstSliceOrigin;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment