Skip to content

Commit

Permalink
Use custom impl and time it
Browse files Browse the repository at this point in the history
  • Loading branch information
EricLBuehler committed Mar 12, 2024
1 parent 8d9218d commit e7c0b9e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion candle-nn/src/layer_norm.rs
Original file line number Diff line number Diff line change
@@ -252,7 +252,7 @@ impl crate::Module for LayerNorm {
.as_micros();
let res=
x.apply_op1_no_bwd(&InnerLayerNorm { eps: self.eps as f32 })?.broadcast_mul(&self.weight)?;
let res = match self.bias {
let res = match &self.bias {
Some(bias) => res.broadcast_add(&bias),
None => Ok(res),
};

0 comments on commit e7c0b9e

Please sign in to comment.