diff options
| author | Lionel Touati <ltouati@gmail.com> | 2024-06-02 14:30:06 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-02 14:30:06 +0200 |
| commit | 1ec3b2cc189fa6020018f2c8dad7b216b4512019 (patch) | |
| tree | fc643da98f7649780798a2668279d39e3441c47f /candle-nn | |
| parent | f7773d498a58fc5678784bd4843011974e11f953 (diff) | |
| download | candle-1ec3b2cc189fa6020018f2c8dad7b216b4512019.tar.gz candle-1ec3b2cc189fa6020018f2c8dad7b216b4512019.tar.bz2 candle-1ec3b2cc189fa6020018f2c8dad7b216b4512019.zip | |
add where_cond f32 for metal (#2236)
Diffstat (limited to 'candle-nn')
| -rw-r--r-- | candle-nn/benches/benchmarks/layer_norm.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-nn/benches/benchmarks/layer_norm.rs b/candle-nn/benches/benchmarks/layer_norm.rs index 0be5c450..4a5fe667 100644 --- a/candle-nn/benches/benchmarks/layer_norm.rs +++ b/candle-nn/benches/benchmarks/layer_norm.rs @@ -5,7 +5,7 @@ use criterion::{black_box, criterion_group, Criterion}; use std::time::Instant; fn run(input: &Tensor, weight: &Tensor, bias: &Tensor) { - let _ = LayerNorm::new(weight.clone(), bias.clone(), 1e-5).forward(&input); + let _ = LayerNorm::new(weight.clone(), bias.clone(), 1e-5).forward(input); } const B: usize = 1; |