Make MLX build on x64 macOS (#901)

The arm64 macbook pros are heavy and I usually care my intel one for
mobile, it would be nice if I can play with MLX on it.

To build with x64, user must pass `MLX_ENABLE_X64_MAC` to cmake:
CMAKE_ARGS='-DMLX_ENABLE_X64_MAC=ON' python setup.py
This commit is contained in:
Cheng
2024-03-27 22:14:29 +09:00
committed by GitHub
parent 90dfa43ff1
commit f30b659291
4 changed files with 40 additions and 30 deletions

View File

@@ -11,8 +11,8 @@ import numpy as np
class MLXTestCase(unittest.TestCase):
@property
def is_linux(self):
return platform.system() == "Linux"
def is_apple_silicon(self):
return platform.machine() == "arm64" and platform.system() == "Darwin"
def setUp(self):
self.default = mx.default_device()