Skip to content

Commit c06b6e4

Browse files
malfetfacebook-github-bot
authored andcommitted
Build M1 binaries with PocketFFT (pytorch#62222)
Summary: As MKL is only available on x86_64 platform, clone header-only PocketFFT library and use it as FFT provider Fixes pytorch#62107 Pull Request resolved: pytorch#62222 Reviewed By: ejguan Differential Revision: D29938718 Pulled By: malfet fbshipit-source-id: ac0bd98b5090d6c8a26c36c4e34a4d6e1d9f1a92
1 parent cb2b5f0 commit c06b6e4

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.jenkins/pytorch/macos-build.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,10 @@ fi
3030
if [ -z "${CROSS_COMPILE_ARM64}" ]; then
3131
USE_DISTRIBUTED=1 python setup.py install
3232
else
33+
pushd "${HOME}"
34+
git clone https://github.com/malfet/pocketfft --depth 1 --branch cpp
35+
popd
36+
export POCKETFFT_HOME="${HOME}/pocketfft"
3337
export MACOSX_DEPLOYMENT_TARGET=11.0
3438
USE_DISTRIBUTED=1 CMAKE_OSX_ARCHITECTURES=arm64 USE_MKLDNN=OFF USE_NNPACK=OFF USE_QNNPACK=OFF BUILD_TEST=OFF python setup.py bdist_wheel
3539
fi

0 commit comments

Comments
 (0)