huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 18:28:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

add where_cond f32 for metal (#2236)

Browse Source
This commit is contained in:
Lionel Touati
2024-06-02 14:30:06 +02:00
committed by GitHub
parent f7773d498a
commit 1ec3b2cc18
3 changed files with 23 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-nn/benches/benchmarks/layer_norm.rs
View File

@ -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;