diff --git a/be/src/util/simd/vstring_function.h b/be/src/util/simd/vstring_function.h index 579da50d2df230..4fff59a01df2d8 100644 --- a/be/src/util/simd/vstring_function.h +++ b/be/src/util/simd/vstring_function.h @@ -17,11 +17,7 @@ #pragma once -#ifdef __AVX2__ #include - -#include "gutil/macros.h" -#endif #include #include @@ -116,7 +112,7 @@ class VStringFunctions { if constexpr (trim_single) { const auto ch = remove_str.data[0]; -#if defined(__AVX2__) +#if defined(__AVX2__) || defined(__aarch64__) constexpr auto AVX2_BYTES = sizeof(__m256i); const auto size = end - begin; const auto* const avx2_begin = end - size / AVX2_BYTES * AVX2_BYTES; @@ -157,7 +153,7 @@ class VStringFunctions { if constexpr (trim_single) { const auto ch = remove_str.data[0]; -#if defined(__AVX2__) +#if defined(__AVX2__) || defined(__aarch64__) constexpr auto AVX2_BYTES = sizeof(__m256i); const auto size = end - begin; const auto* const avx2_end = begin + size / AVX2_BYTES * AVX2_BYTES;