mirror/eigen
2
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-03-07 18:27:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Convert StridedLinearBufferCopy::Kind to enum class

Browse Source
This commit is contained in:
Eugene Zhulenev 2020-01-13 11:43:24 -08:00
parent 5a8b97b401
commit b9362fb8f7
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
unsupported/Eigen/CXX11/src/Tensor/TensorBlock.h
View File

@ -999,7 +999,7 @@ class StridedLinearBufferCopy {
public:
// Specifying linear copy kind statically gives ~30% speedup for small sizes.
enum Kind {
enum class Kind {
Linear = 0, // src_stride == 1 && dst_stride == 1
Scatter = 1, // src_stride == 1 && dst_stride != 1
FillLinear = 2, // src_stride == 0 && dst_stride == 1
@ -1053,7 +1053,7 @@ class StridedLinearBufferCopy {
const IndexType vectorized_size = count - PacketSize;
IndexType i = 0;
if (kind == Linear) {
if (kind == StridedLinearBufferCopy::Kind::Linear) {
// ******************************************************************** //
// Linear copy from `src` to `dst`.
const IndexType unrolled_size = count - 4 * PacketSize;
@ -1072,7 +1072,7 @@ class StridedLinearBufferCopy {
dst[i] = src[i];
}
// ******************************************************************** //
} else if (kind == Scatter) {
} else if (kind == StridedLinearBufferCopy::Kind::Scatter) {
// Scatter from `src` to `dst`.
eigen_assert(src_stride == 1 && dst_stride != 1);
for (; i <= vectorized_size; i += PacketSize) {
@ -1083,7 +1083,7 @@ class StridedLinearBufferCopy {
dst[i * dst_stride] = src[i];
}
// ******************************************************************** //
} else if (kind == FillLinear) {
} else if (kind == StridedLinearBufferCopy::Kind::FillLinear) {
// Fill `dst` with value at `*src`.
eigen_assert(src_stride == 0 && dst_stride == 1);
const IndexType unrolled_size = count - 4 * PacketSize;
@ -1100,7 +1100,7 @@ class StridedLinearBufferCopy {
dst[i] = *src;
}
// ******************************************************************** //
} else if (kind == FillScatter) {
} else if (kind == StridedLinearBufferCopy::Kind::FillScatter) {
// Scatter `*src` into `dst`.
eigen_assert(src_stride == 0 && dst_stride != 1);
Packet p = pload1<Packet>(src);
@ -1111,7 +1111,7 @@ class StridedLinearBufferCopy {
dst[i * dst_stride] = *src;
}
// ******************************************************************** //
} else if (kind == Gather) {
} else if (kind == StridedLinearBufferCopy::Kind::Gather) {
// Gather from `src` into `dst`.
eigen_assert(dst_stride == 1);
for (; i <= vectorized_size; i += PacketSize) {
@ -1122,7 +1122,7 @@ class StridedLinearBufferCopy {
dst[i] = src[i * src_stride];
}
// ******************************************************************** //
} else if (kind == Random) {
} else if (kind == StridedLinearBufferCopy::Kind::Random) {
// Random.
for (; i < count; ++i) {
dst[i * dst_stride] = src[i * src_stride];
@ -1300,17 +1300,17 @@ class TensorBlockIO {
return num_copied;
if (input_stride == 1 && output_stride == 1) {
COPY_INNER_DIM(LinCopy::Linear);
COPY_INNER_DIM(LinCopy::Kind::Linear);
} else if (input_stride == 1 && output_stride != 1) {
COPY_INNER_DIM(LinCopy::Scatter);
COPY_INNER_DIM(LinCopy::Kind::Scatter);
} else if (input_stride == 0 && output_stride == 1) {
COPY_INNER_DIM(LinCopy::FillLinear);
COPY_INNER_DIM(LinCopy::Kind::FillLinear);
} else if (input_stride == 0 && output_stride != 1) {
COPY_INNER_DIM(LinCopy::FillScatter);
COPY_INNER_DIM(LinCopy::Kind::FillScatter);
} else if (output_stride == 1) {
COPY_INNER_DIM(LinCopy::Gather);
COPY_INNER_DIM(LinCopy::Kind::Gather);
} else {
COPY_INNER_DIM(LinCopy::Random);
COPY_INNER_DIM(LinCopy::Kind::Random);
}
#undef COPY_INNER_DIM