experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Ruy: Clean up x86 packing profiling labels.

Browse Source 
PiperOrigin-RevId: 277081266
Change-Id: I9b76bcb3de76eaa10f2d4e9c4d0e076a2089dbdd
This commit is contained in:
Alex Stark 2019-10-28 09:39:32 -07:00 committed by TensorFlower Gardener
parent e55d2881b6
commit 09a394544e
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/experimental/ruy/pack_x86.h
View File

@ -121,7 +121,7 @@ struct PackImpl<Path::kAvx2, FixedKernelLayout<Order::kColMajor, 4, 8>, Scalar,
static void Run(Tuning tuning, const Matrix<Scalar>& src_matrix,
PackedMatrix<std::int8_t>* packed_matrix, int start_col,
int end_col) {
gemmlowp::ScopedProfilingLabel label("Pack (AVX-512)");
gemmlowp::ScopedProfilingLabel label("Pack (AVX2 8-bit)");
RUY_DCHECK(IsColMajor(src_matrix.layout));
RUY_DCHECK(IsColMajor(packed_matrix->layout));
@ -160,6 +160,7 @@ struct PackImpl<Path::kAvx2, FixedKernelLayout<Order::kRowMajor, 1, 8>, float,
static void Run(Tuning, const Matrix<float>& src_matrix,
PackedMatrix<float>* packed_matrix, int start_col,
int end_col) {
gemmlowp::ScopedProfilingLabel label("Pack (AVX2 float)");
RUY_DCHECK(IsColMajor(src_matrix.layout));
RUY_DCHECK(IsColMajor(packed_matrix->layout));
RUY_DCHECK_EQ((end_col - start_col) % Layout::kCols, 0);
@ -204,7 +205,7 @@ struct PackImpl<Path::kAvx512, FixedKernelLayout<Order::kColMajor, 4, 16>,
static void Run(Tuning tuning, const Matrix<Scalar>& src_matrix,
PackedMatrix<std::int8_t>* packed_matrix, int start_col,
int end_col) {
gemmlowp::ScopedProfilingLabel label("Pack (AVX-512)");
gemmlowp::ScopedProfilingLabel label("Pack (AVX-512 8-bit)");
RUY_DCHECK(IsColMajor(src_matrix.layout));
RUY_DCHECK(IsColMajor(packed_matrix->layout));
@ -243,6 +244,7 @@ struct PackImpl<Path::kAvx512, FixedKernelLayout<Order::kRowMajor, 1, 16>,
static void Run(Tuning, const Matrix<float>& src_matrix,
PackedMatrix<float>* packed_matrix, int start_col,
int end_col) {
gemmlowp::ScopedProfilingLabel label("Pack (AVX-512 float)");
using Layout = FixedKernelLayout<Order::kRowMajor, 1, 16>;
RUY_DCHECK(IsColMajor(src_matrix.layout));
RUY_DCHECK(IsColMajor(packed_matrix->layout));