diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-10-06 23:24:55 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-10-06 23:24:55 +0200 |
| commit | e4a96f9e7c2b88dec33b6076cc9756ac76d44df1 (patch) | |
| tree | df5d5f051c9cc22b46f9f2a26abd7f0e82fa2a07 /candle-core/src/metal_backend/mod.rs | |
| parent | f856b5c3a75028d384c26e36501d429091662cd3 (diff) | |
| download | candle-e4a96f9e7c2b88dec33b6076cc9756ac76d44df1.tar.gz candle-e4a96f9e7c2b88dec33b6076cc9756ac76d44df1.tar.bz2 candle-e4a96f9e7c2b88dec33b6076cc9756ac76d44df1.zip | |
Switch to using the MLX matmul by default. (#2547)
Diffstat (limited to 'candle-core/src/metal_backend/mod.rs')
| -rw-r--r-- | candle-core/src/metal_backend/mod.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/candle-core/src/metal_backend/mod.rs b/candle-core/src/metal_backend/mod.rs index 6f560c02..34931c9d 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, |