Skip to content

Commit

Permalink
Fixed
Browse files Browse the repository at this point in the history
  • Loading branch information
EricLBuehler committed Jan 11, 2025
1 parent e93188f commit 884167a
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion candle-nn/src/ops.rs
Original file line number Diff line number Diff line change
Expand Up @@ -717,7 +717,6 @@ impl candle::InplaceOp2 for AttnSoftmaxLastDim {
block_dim: (nth as u32, 1, 1),
shared_mem_bytes: (WARP_SIZE * std::mem::size_of::<f32>()) as u32,
};
dbg!(&cfg);
let func =
dev.get_or_load_func(&kernel_name::<T>("attn_soft_max"), kernels::REDUCE)?;
let params = (&a, &mask, &a, ncols_x as i32, nrows_y as i32, self.scale);
Expand Down

0 comments on commit 884167a

Please sign in to comment.