diff options
| author | Nicolas Patry <patry.nicolas@protonmail.com> | 2023-12-15 11:24:47 +0100 |
|---|---|---|
| committer | Nicolas Patry <patry.nicolas@protonmail.com> | 2023-12-15 11:24:47 +0100 |
| commit | 26540641c1f0a7b351f5e3d3c3c165221ae1d9ed (patch) | |
| tree | 8cebb4af853fad249fd619236d271af085165a78 /candle-nn/src | |
| parent | 34d83377f6e63ef428c82448515dbba0047fdfae (diff) | |
| download | candle-26540641c1f0a7b351f5e3d3c3c165221ae1d9ed.tar.gz candle-26540641c1f0a7b351f5e3d3c3c165221ae1d9ed.tar.bz2 candle-26540641c1f0a7b351f5e3d3c3c165221ae1d9ed.zip | |
Renamed all kernel names.
Diffstat (limited to 'candle-nn/src')
| -rw-r--r-- | candle-nn/src/ops.rs | 6 |
1 files changed, 3 insertions, 3 deletions
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:?}"), }; |