arrow/util/simd.h

0001 // Licensed to the Apache Software Foundation (ASF) under one
0002 // or more contributor license agreements.  See the NOTICE file
0003 // distributed with this work for additional information
0004 // regarding copyright ownership.  The ASF licenses this file
0005 // to you under the Apache License, Version 2.0 (the
0006 // "License"); you may not use this file except in compliance
0007 // with the License.  You may obtain a copy of the License at
0008 //
0009 //   http://www.apache.org/licenses/LICENSE-2.0
0010 //
0011 // Unless required by applicable law or agreed to in writing,
0012 // software distributed under the License is distributed on an
0013 // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
0014 // KIND, either express or implied.  See the License for the
0015 // specific language governing permissions and limitations
0016 // under the License.
0017
0018 #pragma once
0019
0020 #ifdef _MSC_VER
0021 // MSVC x86_64/arm64
0022
0023 #  if defined(_M_AMD64) || defined(_M_X64)
0024 #    include <intrin.h>
0025 #  endif
0026
0027 #else
0028 // gcc/clang (possibly others)
0029
0030 #  if defined(ARROW_HAVE_BMI2) || defined(ARROW_HAVE_RUNTIME_BMI2)
0031 #    include <x86intrin.h>
0032 #  endif
0033
0034 #  if defined(ARROW_HAVE_AVX2) || defined(ARROW_HAVE_AVX512) || \
0035       defined(ARROW_HAVE_RUNTIME_AVX2) || defined(ARROW_HAVE_RUNTIME_AVX512)
0036 #    include <immintrin.h>
0037 #  elif defined(ARROW_HAVE_SSE4_2) || defined(ARROW_HAVE_RUNTIME_SSE4_2)
0038 #    include <nmmintrin.h>
0039 #  endif
0040
0041 #  ifdef ARROW_HAVE_NEON
0042 #    include <arm_neon.h>
0043 #  endif
0044
0045 // GH-44098: Workaround for missing _mm256_set_m128i in older versions of GCC.
0046 #  if defined(__GNUC__) && !defined(__clang__) && __GNUC__ < 8
0047 #    define _mm256_set_m128i(hi, lo) \
0048       _mm256_inserti128_si256(_mm256_castsi128_si256(lo), (hi), 1)
0049 #  endif
0050
0051 #endif