Skip to content

Commit 8675135

Browse files
committed
Force numpy use of Accelerate.framework on macOS
1 parent 4123cd8 commit 8675135

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

edgedbpkg/edgedb/__init__.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -280,6 +280,7 @@ def get_package_repository(
280280
repo.register_package_impl("gel", EdgeDBPython)
281281
repo.register_package_impl("maturin", Maturin)
282282
repo.register_package_impl("cython", Cython)
283+
repo.register_package_impl("numpy", NumPy)
283284
return repo
284285

285286
@property
@@ -750,6 +751,23 @@ class EdgeDB(EdgeDBNoPostgres):
750751
)
751752

752753

754+
class NumPy(packages.PythonPackage):
755+
def sh_get_build_wheel_env(
756+
self, build: targets.Build, *, site_packages: str, wd: str
757+
) -> packages.Args:
758+
env = super().sh_get_build_wheel_env(
759+
build, site_packages=site_packages, wd=wd
760+
)
761+
762+
if build.target.ident.startswith("macos"):
763+
env |= {
764+
"NPY_LAPACK_ORDER": "accelerate",
765+
"NPY_BLAS_ORDER": "accelerate",
766+
}
767+
768+
return env
769+
770+
753771
class Cryptography(packages.PythonPackage):
754772
def sh_get_build_wheel_env(
755773
self, build: targets.Build, *, site_packages: str, wd: str

0 commit comments

Comments
 (0)