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

Fix compile error (use of avx512 intrinsics without including header)

Browse Source 
and wrong opt set (RUY_OPT_INTRINSICS, not RUY_OPT_ASM, there is no asm here).

PiperOrigin-RevId: 261314107
This commit is contained in:
Benoit Jacob 2019-08-02 06:57:40 -07:00 committed by TensorFlower Gardener
parent e9b0a23e6e
commit 555de8b4c3
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/lite/experimental/ruy/pack_avx512.cc
View File

@ -16,9 +16,13 @@ limitations under the License.
#include "tensorflow/lite/experimental/ruy/pack.h"
#include "tensorflow/lite/experimental/ruy/platform.h"
#if RUY_PLATFORM(AVX512) && RUY_OPT_ENABLED(RUY_OPT_INTRINSICS)
#include <immintrin.h>
#endif
namespace ruy {
#if RUY_PLATFORM(AVX512) && RUY_OPT_ENABLED(RUY_OPT_ASM)
#if RUY_PLATFORM(AVX512) && RUY_OPT_ENABLED(RUY_OPT_INTRINSICS)
// The first int8_t template parameter is arbitrary: this routine is common to
// all 8-bit source matrix types.
@ -526,6 +530,6 @@ void PackFloatAvx512(const float* src_ptr, const float* zerobuf, int src_stride,
}
}
#endif // RUY_PLATFORM(AVX512) && RUY_OPT_ENABLED(RUY_OPT_ASM)
#endif // RUY_PLATFORM(AVX512) && RUY_OPT_ENABLED(RUY_OPT_INTRINSICS)
} // namespace ruy