Skip to content

Commit c0dc6c5

Browse files
authored
GH-49327: [Python][Packaging] Use ARROW_SIMD_LEVEL=NEON for macOS arm64 (#50181)
### Rationale for this change Currently macOS arm64 wheels are trying to set `ARROW_SIMD_LEVEL=SSE4_2` ### What changes are included in this PR? Set `ARROW_SIMD_LEVEL` to `SSE4_2` or `NEON` based on architecture ### Are these changes tested? Via archery ### Are there any user-facing changes? No but macOS wheels will use the SIMD optimizations. * GitHub Issue: #49327 Authored-by: Raúl Cumplido <raulcumplido@gmail.com> Signed-off-by: Sutou Kouhei <kou@clear-code.com>
1 parent 9a7e801 commit c0dc6c5

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

ci/scripts/python_wheel_macos_build.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,10 @@ export SDKROOT=${SDKROOT:-$(xcrun --sdk macosx --show-sdk-path)}
3939

4040
if [ $arch = "arm64" ]; then
4141
export CMAKE_OSX_ARCHITECTURES="arm64"
42+
: ${ARROW_SIMD_LEVEL:="NEON"}
4243
elif [ $arch = "x86_64" ]; then
4344
export CMAKE_OSX_ARCHITECTURES="x86_64"
45+
: ${ARROW_SIMD_LEVEL:="SSE4_2"}
4446
else
4547
echo "Unexpected architecture: $arch"
4648
exit 1
@@ -68,7 +70,6 @@ echo "=== (${PYTHON_VERSION}) Building Arrow C++ libraries ==="
6870
: ${PARQUET_REQUIRE_ENCRYPTION:=ON}
6971
: ${ARROW_SUBSTRAIT:=ON}
7072
: ${ARROW_S3:=ON}
71-
: ${ARROW_SIMD_LEVEL:="SSE4_2"}
7273
: ${ARROW_TENSORFLOW:=ON}
7374
: ${ARROW_WITH_BROTLI:=ON}
7475
: ${ARROW_WITH_BZ2:=ON}

0 commit comments

Comments
 (0)