Skip to content

Commit

Permalink
fix typo (#2363)
Browse files Browse the repository at this point in the history
  • Loading branch information
tiruka authored Oct 15, 2024
1 parent a5136d4 commit c38f212
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion crates/burn-jit/src/kernel/conv/conv2d/implicit_gemm.rs
Original file line number Diff line number Diff line change
Expand Up @@ -472,7 +472,7 @@ fn load_input_tile<F: Float, FMat: Float>(
// Slices are always `kernel_size * channels` elements wide so we can compute where inside a slice
// we are and also which row the slice is in relative to the start of the CMMA matrix

// Actual index within a slice (0 to `kernel_size * channels - 1`) that the thread is repsonsible for
// Actual index within a slice (0 to `kernel_size * channels - 1`) that the thread is responsible for
let my_slice_idx = (slice_start_idx + (m % cmma_k)) % dims.slice_size;

let channel = my_slice_idx % channels;
Expand Down

0 comments on commit c38f212

Please sign in to comment.