From 1ec3b2cc189fa6020018f2c8dad7b216b4512019 Mon Sep 17 00:00:00 2001 From: Lionel Touati Date: Sun, 2 Jun 2024 14:30:06 +0200 Subject: add where_cond f32 for metal (#2236) --- candle-nn/benches/benchmarks/layer_norm.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'candle-nn') 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; -- cgit v1.2.3