dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

merged

This commit is contained in:
louisfd 2024-06-28 11:12:59 -04:00
parent 0dbc3c5af4 61ca9ff0b6
commit 97b53624b9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
crates/burn-jit/src/kernel/matmul/tiling2d_cube/load_shared_memory.rs
View File

@ -200,7 +200,7 @@ fn write_tile_transposed<F: Float>(
} else {
for i in range(0u32, Comptime::get(tile_size), unroll) {
let mut transposed = F::vectorized(0., Comptime::get(tile_size));
for j in range(0u32, Comptime::get(tile_size), unroll) {
for j in range(0u32, Comptime::get(tile_size), Comptime::new(true)) {
let mut row = tile[j];
if j > Comptime::runtime(tile_size) {
row = F::vectorized(0., Comptime::get(tile_size));