Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Eps double -> float
Browse files Browse the repository at this point in the history
EricLBuehler committed Mar 12, 2024
1 parent 4bf0f7c commit 6cf2143
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions candle-nn/src/layer_norm.rs
Original file line number Diff line number Diff line change
@@ -209,6 +209,7 @@ impl crate::Module for LayerNorm {
None => candle::bail!("input has to be contiguous"),
Some((o1, o2)) => slice.slice(o1..o2),
};
dbg(T);
let elem_count = layout.shape().elem_count();
let dst = unsafe { dev.alloc::<T>(elem_count) }.w()?;
let func = dev.get_or_load_func(

0 comments on commit 6cf2143

Please sign in to comment.