v4.0 update. (#2398)

* Ex77 fix.
2025-06-12 09:10:29 -04:00
parent 5c6bca0441
commit dc4817921e
4 changed files with 142 additions and 27 deletions
@@ -403,6 +403,7 @@ struct Sm100FmhaFwdKernelTmaWarpspecialized {
        mainloop.correction(
          blk_coord,
          params.mainloop, logical_problem_shape,
+          params.problem_shape,
          shared_storage.epilogue,
          pipeline_s0_corr, pipeline_s0_corr_consumer_state,
          pipeline_s1_corr, pipeline_s1_corr_consumer_state,