diff --git a/source/backend/cpu/bf16/VecHalf.hpp b/source/backend/cpu/bf16/VecHalf.hpp index 4568f95a..ee93b487 100644 --- a/source/backend/cpu/bf16/VecHalf.hpp +++ b/source/backend/cpu/bf16/VecHalf.hpp @@ -12,6 +12,18 @@ #include #include #include // supply std::max and std::min + +#ifdef MNN_USE_NEON +#include +#endif +#ifdef MNN_USE_SSE +#if defined(_MSC_VER) +#include +#else +#include +#endif +#endif + namespace MNN { namespace Math { @@ -160,11 +172,6 @@ struct VecHalf { }; #if defined(MNN_USE_SSE) -#if defined(_MSC_VER) -#include -#else -#include -#endif template<> struct VecHalf<4> { @@ -310,7 +317,6 @@ struct VecHalf<4> { #endif #if defined(MNN_USE_NEON) -#include template<> struct VecHalf<4> {