From 35c87741cf2450c96c6e52afead61eec81c45e2a Mon Sep 17 00:00:00 2001 From: Cheng Date: Wed, 21 May 2025 11:42:48 +0900 Subject: [PATCH] Build for compute capability 70 instead of 75 (#2209) --- mlx/backend/cuda/CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlx/backend/cuda/CMakeLists.txt b/mlx/backend/cuda/CMakeLists.txt index f9695f66a..d62a69846 100644 --- a/mlx/backend/cuda/CMakeLists.txt +++ b/mlx/backend/cuda/CMakeLists.txt @@ -25,7 +25,7 @@ target_compile_options(mlx # Compute capability 7 is required for synchronization between CPU/GPU with # managed memory. TODO: Add more architectures for potential performance gain. set(MLX_CUDA_ARCHITECTURES - "75;80" + "70;80" CACHE STRING "CUDA architectures") message(STATUS "CUDA architectures: ${MLX_CUDA_ARCHITECTURES}") set_target_properties(mlx PROPERTIES CUDA_ARCHITECTURES