From d3b1b2d0fbf60e98f2b2243814d72d78c8311741 Mon Sep 17 00:00:00 2001 From: Vlad Severin Date: Sun, 2 Mar 2025 16:54:55 +0200 Subject: [PATCH] make bbox build work on arm mac --- widerface_evaluate/setup.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/widerface_evaluate/setup.py b/widerface_evaluate/setup.py index 74dba05b3..d477dab25 100644 --- a/widerface_evaluate/setup.py +++ b/widerface_evaluate/setup.py @@ -8,6 +8,23 @@ from distutils.core import setup, Extension from Cython.Build import cythonize import numpy +import platform +import os -package = Extension('bbox', ['box_overlaps.pyx'], include_dirs=[numpy.get_include()]) +extra_compile_args = [] +extra_link_args = [] + +if platform.system() == 'Darwin' and platform.machine() == 'arm64': + os.environ["CFLAGS"] = "-arch arm64" + os.environ["ARCHFLAGS"] = "-arch arm64" + extra_compile_args.extend(["-arch", "arm64"]) + extra_link_args.extend(["-arch", "arm64"]) + +package = Extension( + 'bbox', + ['box_overlaps.pyx'], + include_dirs=[numpy.get_include()], + extra_compile_args=extra_compile_args, + extra_link_args=extra_link_args +) setup(ext_modules=cythonize([package]))