From 018bda719f675b127bd3398ba684175ca425cc97 Mon Sep 17 00:00:00 2001 From: Eric Buehler Date: Thu, 31 Oct 2024 08:57:26 -0400 Subject: [PATCH] Some review comments --- .vscode/settings.json | 1 - candle-nn/src/ops.rs | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/.vscode/settings.json b/.vscode/settings.json index 383df783ce..36c8eb1c90 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 a57b988e03..15a5fbec70 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(