diff --git a/.vscode/settings.json b/.vscode/settings.json index 383df783c..36c8eb1c9 100644 --- a/.vscode/settings.json +++ b/.vscode/settings.json @@ -8,5 +8,4 @@ ], "python.testing.unittestEnabled": false, "python.testing.pytestEnabled": true, - "rust-analyzer.cargo.features": ["metal"] } \ No newline at end of file diff --git a/candle-nn/src/ops.rs b/candle-nn/src/ops.rs index a57b988e0..15a5fbec7 100644 --- a/candle-nn/src/ops.rs +++ b/candle-nn/src/ops.rs @@ -956,7 +956,7 @@ struct Sdpa { impl candle::CustomOp3 for Sdpa { fn name(&self) -> &'static str { - "sdpa" + "metal-sdpa" } fn cpu_fwd(