From 5900e3249f8b48459b19d84cb146590a122ed387 Mon Sep 17 00:00:00 2001 From: Cheng Date: Mon, 30 Sep 2024 23:00:39 +0900 Subject: [PATCH] Fix building on Linux (#1446) --- CMakeLists.txt | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index b9594921d..1688b2052 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -14,7 +14,7 @@ option(MLX_BUILD_TESTS "Build tests for mlx" ON) option(MLX_BUILD_EXAMPLES "Build examples for mlx" ON) option(MLX_BUILD_BENCHMARKS "Build benchmarks for mlx" OFF) option(MLX_BUILD_PYTHON_BINDINGS "Build python bindings for mlx" OFF) -option(MLX_BUILD_METAL "Build metal backend" ON) +option(MLX_BUILD_METAL "Build metal backend" OFF) option(MLX_BUILD_CPU "Build cpu backend" ON) option(MLX_METAL_DEBUG "Enhance metal debug workflow" OFF) option(MLX_ENABLE_X64_MAC "Enable building for x64 macOS" OFF) @@ -49,15 +49,18 @@ if(${CMAKE_SYSTEM_NAME} MATCHES "Darwin") else() message(WARNING "Building for x86_64 arch is not officially supported.") endif() - set(MLX_BUILD_METAL OFF) elseif(${CMAKE_SYSTEM_PROCESSOR} MATCHES "arm64") - set(MLX_BUILD_ARM ON) + set(MLX_BUILD_METAL ON) endif() else() message(WARNING "MLX is prioritised for Apple silicon systems using macOS.") endif() +if(${CMAKE_SYSTEM_PROCESSOR} MATCHES "arm64") + set(MLX_BUILD_ARM ON) +endif() + # ----------------------------- Lib ----------------------------- include(FetchContent)