From 26540641c1f0a7b351f5e3d3c3c165221ae1d9ed Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Fri, 15 Dec 2023 11:24:47 +0100
Subject: Renamed all kernel names.

---
 candle-nn/src/ops.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'candle-nn/src')

diff --git a/candle-nn/src/ops.rs b/candle-nn/src/ops.rs
index f00d8e2f..ca23f90e 100644
--- a/candle-nn/src/ops.rs
+++ b/candle-nn/src/ops.rs
@@ -213,9 +213,9 @@ impl candle::CustomOp1 for SoftmaxLastDim {
         let command_buffer = device.command_buffer();
         let kernels = device.kernels();
         let name = match storage.dtype() {
-            DType::F32 => "softmax_float",
-            DType::F16 => "softmax_half",
-            DType::BF16 => "softmax_bfloat",
+            DType::F32 => "softmax_f32",
+            DType::F16 => "softmax_f16",
+            DType::BF16 => "softmax_bf16",
             dtype => candle::bail!("softmax-last-dim is not implemented for {dtype:?}"),
         };
 
-- 
cgit v1.2.3