From e4a96f9e7c2b88dec33b6076cc9756ac76d44df1 Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Sun, 6 Oct 2024 23:24:55 +0200 Subject: [PATCH] Switch to using the MLX matmul by default. (#2547) --- candle-core/src/metal_backend/mod.rs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/candle-core/src/metal_backend/mod.rs b/candle-core/src/metal_backend/mod.rs index 6f560c02ee..34931c9dfd 100644 --- a/candle-core/src/metal_backend/mod.rs +++ b/candle-core/src/metal_backend/mod.rs @@ -1865,9 +1865,9 @@ impl BackendDevice for MetalDevice { let device = metal::Device::all().swap_remove(ordinal); let command_queue = device.new_command_queue(); let kernels = Arc::new(Kernels::new()); - let use_mlx_mm = match std::env::var("CANDLE_USE_MLX_MM").as_deref() { - Ok("false") | Ok("False") | Ok("FALSE") | Ok("0") | Err(_) => false, - Ok(_) => true, + let use_mlx_mm = match std::env::var("CANDLE_USE_MFA_MM").as_deref() { + Ok("false") | Ok("False") | Ok("FALSE") | Ok("0") | Err(_) => true, + Ok(_) => false, }; let seed = Arc::new(Mutex::new(device.new_buffer_with_data( [299792458].as_ptr() as *const c_void,