Skip to content

Commit 9a42c88

Browse files
committed
update dense gemm source code
1 parent c1bcfcf commit 9a42c88

2 files changed

Lines changed: 1 addition & 2 deletions

File tree

-2 Bytes
Binary file not shown.

‎optimization/dense-gemm-fp16-from-naive.md‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,6 @@ __co__ void matmul(
326326
shared event full[STAGES], empty[STAGES];
327327
shared f16 [TILE_M, TILE_K] lhs_s[STAGES];
328328
shared f16 [WARP_N, TILE_K] rhs_s[STAGES];
329-
shared f16 [WARP_M, WARP_N] output_s[CONSUMERS];
330329
331330
// 3 warpgroups × 128 threads = 384 threads/block
332331
parallel wg by 3 : group-4, t by 128 : thread {
@@ -362,7 +361,7 @@ __co__ void matmul(
362361
}
363362
trigger empty[stage];
364363
}
365-
364+
shared f16 [WARP_M, WARP_N] output_s[CONSUMERS];
366365
mma.store mc, output_s[cidx];
367366
tma.copy output_s[cidx] =>
368367
output.subspan(WARP_M, WARP_N).at(block_m * CONSUMERS + cidx, block_n);

0 commit comments

Comments
 (0)