Skip to content

Commit

Permalink
Format
Browse files Browse the repository at this point in the history
  • Loading branch information
EricLBuehler committed Jan 14, 2025
1 parent 58b6a21 commit 900ca40
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion candle-metal-kernels/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -757,7 +757,7 @@ pub fn call_last_attn_softmax(
let el: usize = input_shape.iter().product();
el / bs
};

let mut nth = 32; // SIMD width
let name = if ne00 % 4 == 0 {
while nth < ne00 / 4 && nth * ne01 * ne02 * ne03 < 256 {
Expand Down

0 comments on commit 900ca40

Please sign in to comment.