Fix out-of-bounds TMA access in wgmma_tma_sm90 tutorial (#2945)

2026-01-22 20:54:12 -08:00
parent 431d070fcb
commit 0edaa6e47d
1 changed files with 2 additions and 1 deletions
--- a/examples/cute/tutorial/hopper/wgmma_tma_sm90.cu
+++ b/examples/cute/tutorial/hopper/wgmma_tma_sm90.cu
@@ -236,7 +236,8 @@ gemm_device(ProblemShape shape_MNK, CtaTiler cta_tiler,
    ConsumerBarType::arrive(&consumer_mbar[read_pipe]);
    ++read_state;

-    if ((warp_idx == 0) && lane_predicate)
+    // Only issue new TMA copies if there are more tiles to fetch
+    if ((warp_idx == 0) && lane_predicate && (k_tile_count > 0))
    {
      int pipe = write_state.index();
      // Wait for Consumer to complete consumption